Update 03.train_C3D.py

AKASH2907 · May 23, 2020 · b97cacd · b97cacd
1 parent 0cc075c
commit b97cacd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/03.train_C3D.py b/03.train_C3D.py
@@ -305,6 +305,7 @@ def generator_val_batch(val_vid_list, batch_size, num_classes):
             b = (i + 1) * batch_size
             y_test, y_labels = process_batch(
                 val_vid_list[a:b],
+                batch_size,
                 train=False
             )
             x = preprocess(y_test)