added config.yaml

koul · Oct 4, 2022 · 8b25804 · 8b25804
1 parent 00de5b4
commit 8b25804
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/config.yaml b/config.yaml
@@ -1,11 +1,11 @@
-global: 
+global:
     root_dir: ""
     iteration: "library_aditi_balanced_run2"
-    enable_preprocessing: true
+    enable_preprocessing: false
 transformer:
     fps: 5
-    path: "data/library"
-    data_save_file: "data/processed/library" #file path without .npz
+    path: "/content/data/"
+    data_save_file: "/content/processed/library" #file path without .npz
 data:
     HEIGHT: 128
     WIDTH: 128
@@ -26,8 +26,9 @@ trainer:
     num_classes: 3
     model_save_path: "models/"
     model:
-        name: "ConvLSTM"
+        name: "PredRNN"
         convlstm_hidden: 128
         num_conv_lstm_layers: 1
+        decouple_beta: 0.1
         pretrained_path: ""
         optimizer_path: ""