Updated baseline.yaml

PIWM/config/agent/baseline.yaml

@@ -16,16 +16,6 @@ denoiser:
     - 2
     - 2
     - 2
-    # channels:
-    # - 48
-    # - 96
-    # - 192
-    # - 384
-    # channels:
-    # - 64
-    # - 128
-    # - 256
-    # - 512
     channels:
     - 128
     - 256
@@ -34,7 +24,7 @@ denoiser:
     attn_depths:
     - 0
     - 0
-    - 1 #0  20M params
+    - 1
     - 1
 
 upsampler:
@@ -44,7 +34,6 @@ upsampler:
   noise_previous_obs: false
   upsampling_factor: 5
   inner_model:
-
     _target_: models.diffusion.InnerModelConfig
     img_channels: 3
     num_steps_conditioning: 1
@@ -63,8 +52,4 @@ upsampler:
     - 0
     - 0
     - 0
-    - 1
-  
-rew_end_model: null 
-
-actor_critic: null 
+    - 1

PIWM/config/agent/piwm.yaml

@@ -69,4 +69,3 @@ upsampler:
     - 0
     - 0
     - 1
-  

PIWM/config/env/baseline.yaml

@@ -1,9 +1,9 @@
 train:
-  id: csgo
-  # size: [150, 280]
+  id: piwm
   size: [150, 600]
 num_actions: 51
-path_data_low_res: /home/zhexiao/Documents/old_highway_dataset_preprocessed/low_res
-path_data_full_res: /home/zhexiao/Documents/old_highway_dataset_preprocessed/full_res
 
-keymap: csgo
+path_data_low_res: null
+path_data_full_res: null
+
+keymap: piwm

PIWM/config/env/piwm.yaml

@@ -3,7 +3,7 @@ train:
   size: [150, 600]
 num_actions: 51
 
-path_data_low_res: /home/zhexiao/Documents/old_highway_dataset_preprocessed/low_res
-path_data_full_res: /home/zhexiao/Documents/old_highway_dataset_preprocessed/full_res
+path_data_low_res: null
+path_data_full_res: null
 
-keymap: piwm
+keymap: piwm