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