validate with variable batch size

supernnova · Apr 3, 2020 · ef330d1 · ef330d1
1 parent f72df11
commit ef330d1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/supernnova/validation/validate_rnn.py b/supernnova/validation/validate_rnn.py
@@ -133,8 +133,7 @@ def get_predictions(settings, model_file=None):
 
     # Batching stuff together
     num_elem = len(list_data_test)
-    factor = (num_elem // 4) if settings.source_data == "photometry" else num_elem
-    num_batches = num_elem / min(factor,100000)
+    num_batches = num_elem / min(num_elem, settings.batch_size)
     list_batches = np.array_split(np.arange(num_elem), num_batches)
 
     # Prepare output arrays