Commit 5d455729 authored by peastman's avatar peastman
Browse files

Bug fix

parent fec23179
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -487,6 +487,7 @@ class ANIRegression(TensorGraph):

        val = npo[k]
        tensor = g.get_tensor_by_name(k)
        if tensor.dtype != tf.resource:  # workaround for save_counter incorrectly being marked as a trainable variable
          all_ops.append(tf.assign(tensor, val))

      obj.session.run(all_ops)