Settings

harvitronix · Nov 21, 2017 · 0db3656 · 0db3656
1 parent fa9492a
commit 0db3656
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/extract_features.py b/extract_features.py
@@ -19,7 +19,7 @@
 
 # Set defaults.
 seq_length = 40
-class_limit = 2  # Number of classes to extract. Can be 1-101 or None for all.
+class_limit = None  # Number of classes to extract. Can be 1-101 or None for all.
 
 # Get the dataset.
 data = DataSet(seq_length=seq_length, class_limit=class_limit)
@@ -36,7 +36,7 @@
         '-features')  # numpy will auto-append .npy
 
     # Check if we already have it.
-    if os.path.isfile(path):
+    if os.path.isfile(path + '.npy'):
         pbar.update(1)
         continue
 

diff --git a/train.py b/train.py
@@ -86,7 +86,7 @@ def main():
     # model can be one of lstm, lrcn, mlp, conv_3d, c3d
     model = 'lstm'
     saved_model = None  # None or weights file
-    class_limit = 10  # int, can be 1-101 or None
+    class_limit = None  # int, can be 1-101 or None
     seq_length = 40
     load_to_memory = False  # pre-load the sequences into memory
     batch_size = 32