File size: 819 Bytes
7ecb82e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
_target_: agent.AgentConfig
denoiser:
_target_: models.diffusion.DenoiserConfig
sigma_data: 0.5
sigma_offset_noise: 0.3
inner_model:
_target_: models.diffusion.InnerModelConfig
img_channels: 3
num_steps_conditioning: 4
cond_channels: 256
depths: [2,2,2,2]
channels: [64,64,64,64]
attn_depths: [0,0,0,0]
rew_end_model:
_target_: models.rew_end_model.RewEndModelConfig
lstm_dim: 512
img_channels: ${agent.denoiser.inner_model.img_channels}
img_size: ${env.train.size}
cond_channels: 128
depths: [2,2,2,2]
channels: [32,32,32,32]
attn_depths: [0,0,0,0]
actor_critic:
_target_: models.actor_critic.ActorCriticConfig
lstm_dim: 512
img_channels: ${agent.denoiser.inner_model.img_channels}
img_size: ${env.train.size}
channels: [32,32,64,64]
down: [1,1,1,1]
|