bug fix with setting params.test_lstm

fMoW · Oct 25, 2017 · e18ab9d · e18ab9d
1 parent ce3e462
commit e18ab9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/fmowBaseline.py b/code/fmowBaseline.py
@@ -61,7 +61,7 @@ def __init__(self, params=None, argv=None):
             if performAll or arg == '-test_cnn':
                 self.params.test_cnn = True
             if performAll or arg == '-test_lstm':
-                self.params.test_cnn = True
+                self.params.test_lstm = True
 
             if arg == '-nm':
                 self.params.use_metadata = False