PIWM / config /world_model_env /fast.yaml
musictimer's picture
Fix bug 3
a29f249
raw
history blame contribute delete
512 Bytes
_target_: src.envs.WorldModelEnvConfig
horizon: 1000
num_batches_to_preload: 1
diffusion_sampler_next_obs:
_target_: src.models.diffusion.DiffusionSamplerConfig
num_steps_denoising: 6 # Balanced: better quality than 3, faster than 10
sigma_min: 0.002
sigma_max: 5.0
rho: 7
order: 1
diffusion_sampler_upsampling:
_target_: src.models.diffusion.DiffusionSamplerConfig
num_steps_denoising: 4 # Balanced: better quality than 2, faster than 5
sigma_min: 0.002
sigma_max: 5.0
rho: 7
order: 1