Bug fixes

Gear-AI · May 27, 2021 · e5f655c · e5f655c
1 parent 0996f17
commit e5f655c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gearai/ga/tf/keras/__init__.py b/gearai/ga/tf/keras/__init__.py
@@ -38,10 +38,10 @@ def train_step(self,data):
 
         # ACCUMULATING THE BATCH GRADIENTS
         for i in range(len(self.grad_acc)):
-            self.gradient_acc[i].assign_add(gradients[i])
+            self.grad_acc[i].assign_add(gradients[i])
 
 
-        tf.cond(tf.equal(self.n_acum_step, self.n_grads), self.apply_accu_gradients, lambda: None)
+        tf.cond(tf.equal(self.n_acum_step, self.no_grads), self.apply_accu_gradients, lambda: None)
 
         # UPDATING THE METRICS
         self.compiled_metrics.update_state(y, y_pred)