Users/jeff

config.yaml

@@ -1,18 +1,18 @@
-global: 
+global:
     root_dir: ""
-    iteration: "intent_indoor_baseline_2"
-    description: "intent_indoor_baseline full run changed run params"
-    enable_preprocessing: False
+    iteration: "intent_Predrnn_focal_dropout"
+    enable_preprocessing: false
     seed: 8
-    enable_intent: True
-    enable_wandb: True
+    enable_intent: true
+    enable_wandb: true
+    description: "predrnn-focal-dropout0.2 multiply-0.8-1.2 translate-5"
 transformer:
     fps: 2
-    path: "data/"
-    data_save_file: "data/train" #file path without .npz
-    enable_benchmark_test: False
-    test_path: "data/"
-    test_save_file: "data/train" #file path without .npz
+    path: "data/Indoor - Seattle Library"
+    data_save_file: "/content/data/Indoor - train" #file path without .npz
+    enable_benchmark_test: true
+    test_path: "data/test/Mixed"
+    test_save_file: "/content/data/Indoor - test" #file path without .npz
 data:
     HEIGHT: 64
     WIDTH: 64
@@ -21,15 +21,30 @@ data:
     VAL_INTENT: ""
     BENCHMARK_TEST_INTENT: ""
 trainer:
-    BATCH: 64
-    lr: 0.0001
+    patch_size: 4
+    img_channel: 1
+    img_width: 128
+    filter_size: 5
+    stride: 1
+    layer_norm: 0
+    visual: 0
+    BATCH: 8
+    max_lr: 0.01
+    lr: 0.00002
     epochs: 25
-    lambda: 0.01  #L2 regularization #changed from 1e-4
+    lambda: 0.001  #L2 regularization #changed from 1e-4
     num_classes: 3
+    wandb: true
     model_save_path: "models/"
     model:
-        name: "ConvLSTM"
-        convlstm_hidden: "256"
+        name: "PredRNN"
+        focal_loss_gamma: 1
+        convlstm_hidden: 128
         num_conv_lstm_layers: 1
+        conv_dropout: 0
+        decouple_beta: 0.1
         pretrained_path: ""
         optimizer_path: ""
+        scheduler_path: ""
+        lr_after_load: ""
+