PIWM_ckpt / PIWM /config /agent /baseline.yaml
Zhexiao0's picture
Updated baseline.yaml
85fa0ab
raw
history blame contribute delete
897 Bytes
_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:
- 128
- 256
- 512
- 1024
attn_depths:
- 0
- 0
- 1
- 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