Fix bug with large batch sizes.

data61 · Apr 22, 2024 · f86ce7c · f86ce7c
1 parent 0468abb
commit f86ce7c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Compiler/ml.py b/Compiler/ml.py
@@ -3294,6 +3294,7 @@ def __init__(self, n_epochs=1, batch_size=1, program=None):
         self.n_epochs = n_epochs
         self.batch_size = batch_size
         self.program = program
+        Layer.back_batch_size = max(Layer.back_batch_size, batch_size)
 
     def fit(self, X_train, y_train):
         """ Train classifier.