Added baseline ckpt

PIWM/config/agent/baseline.yaml

@@ -0,0 +1,70 @@
+_target_: agent.AgentConfig
+
+denoiser:
+  _target_: models.diffusion.DenoiserConfig
+  sigma_data: 0.5
+  sigma_offset_noise: 0.1
+  noise_previous_obs: true
+  upsampling_factor: null
+  inner_model:
+    _target_: models.diffusion.InnerModelConfig
+    img_channels: 3
+    num_steps_conditioning: 4
+    cond_channels: 2048
+    depths:
+    - 2
+    - 2
+    - 2
+    - 2
+    # channels:
+    # - 48
+    # - 96
+    # - 192
+    # - 384
+    # channels:
+    # - 64
+    # - 128
+    # - 256
+    # - 512
+    channels:
+    - 128
+    - 256
+    - 512
+    - 1024
+    attn_depths:
+    - 0
+    - 0
+    - 1 #0  20M params
+    - 1
+
+upsampler:
+  _target_: models.diffusion.DenoiserConfig
+  sigma_data: 0.5
+  sigma_offset_noise: 0.1
+  noise_previous_obs: false
+  upsampling_factor: 5
+  inner_model:
+
+    _target_: models.diffusion.InnerModelConfig
+    img_channels: 3
+    num_steps_conditioning: 1
+    cond_channels: 2048
+    depths:
+    - 2
+    - 2
+    - 2
+    - 2
+    channels:
+    - 64
+    - 64
+    - 128
+    - 256
+    attn_depths:
+    - 0
+    - 0
+    - 0
+    - 1
+  
+rew_end_model: null 
+
+actor_critic: null 

PIWM/config/env/baseline.yaml

@@ -0,0 +1,9 @@
+train:
+  id: csgo
+  # size: [150, 280]
+  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

PIWM/model/baseline.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c30005e0e46b40af7c0f35de0c8d294c1bf591b5c8f44a8a106a468df0c05b2d
+size 1526816194