change default ft_denoising_steps in eval configs to 0 (assume evaluating pre-trained models)

This commit is contained in:
allenzren
2025-02-04 11:51:47 -05:00
committed by Allen Z. Ren
parent fc42865c77
commit 9032d02eae
28 changed files with 55 additions and 55 deletions
@@ -21,7 +21,7 @@ horizon_steps: 8
act_steps: 8
use_ddim: True
ddim_steps: 5
ft_denoising_steps: 5
ft_denoising_steps: 0
n_steps: ${eval:'round(${env.max_episode_steps} / ${act_steps})'}
render_num: 0
@@ -42,7 +42,7 @@ env:
sparse_reward: True
model:
_target_: model.diffusion.diffusion_eval_ft.DiffusionEvalFT
_target_: model.diffusion.diffusion_eval_ft.DiffusionEval
ft_denoising_steps: ${ft_denoising_steps}
predict_epsilon: True
denoised_clip_value: 1.0
@@ -21,7 +21,7 @@ horizon_steps: 16
act_steps: 8
use_ddim: True
ddim_steps: 5
ft_denoising_steps: 5
ft_denoising_steps: 0
n_steps: ${eval:'round(${env.max_episode_steps} / ${act_steps})'}
render_num: 0
@@ -42,7 +42,7 @@ env:
sparse_reward: True
model:
_target_: model.diffusion.diffusion_eval_ft.DiffusionEvalFT
_target_: model.diffusion.diffusion_eval_ft.DiffusionEval
ft_denoising_steps: ${ft_denoising_steps}
predict_epsilon: True
denoised_clip_value: 1.0