_target_: src.agent.AgentConfig

denoiser:
  _target_: src.models.diffusion.DenoiserConfig
  sigma_data: 0.5
  sigma_offset_noise: 0.1
  noise_previous_obs: true
  upsampling_factor: null
  inner_model:
    _target_: src.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_: src.models.diffusion.DenoiserConfig
  sigma_data: 0.5
  sigma_offset_noise: 0.1
  noise_previous_obs: false
  upsampling_factor: 5
  inner_model:
    _target_: src.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:
  _target_: src.models.actor_critic.ActorCriticConfig
  lstm_dim: 512
  img_channels: 3
  img_size: 64
  channels: [32, 64, 128]
  down: [2, 2, 2]