Update train.py

zishansami102 · Nov 29, 2019 · f8c1581 · f8c1581
1 parent 0689f2d
commit f8c1581
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/train.py b/train.py
@@ -14,11 +14,9 @@
 N_EPOCHS = 2
 REG_PENALTY = 0
 PER=0.2
-# NUM_VID = 500
 NUM_IMAGES = 599900	
 NUM_TEST_IMAGES = 199900
-# NUM_IMAGES = 10000	
-# NUM_TEST_IMAGES = 4000
+