Update m1_tf_test.py

2022-07-11 11:58:04 +02:00
parent 851a2f92b2
commit 5dc0c300e8
1 changed files with 1 additions and 1 deletions
@@ -19,7 +19,7 @@ class ComputeSum(Layer):

  def __init__(self, input_dim):
      super(ComputeSum, self).__init__()
-      # Create a non-trainable weight.
+      # Create a trainable weight.
      self.total = tf.Variable(initial_value=tf.zeros((input_dim,)),
                               trainable=True)