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
@@ -20,7 +20,7 @@ denoising_steps: 20
cond_steps: 1
horizon_steps: 8
act_steps: 8
ft_denoising_steps: 10
ft_denoising_steps: 0
n_steps: 400 # each episode takes max_episode_steps / act_steps steps
render_num: 0
@@ -48,7 +48,7 @@ env:
reset_within_step: 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
@@ -23,7 +23,7 @@ horizon_steps: 8
act_steps: 8
use_ddim: True
ddim_steps: 5
ft_denoising_steps: 5
ft_denoising_steps: 0
n_steps: 200 # each episode takes max_episode_steps / act_steps steps
render_num: 0
@@ -63,7 +63,7 @@ shape_meta:
shape: [14]
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
@@ -20,7 +20,7 @@ denoising_steps: 20
cond_steps: 1
horizon_steps: 16
act_steps: 8
ft_denoising_steps: 10
ft_denoising_steps: 0
n_steps: 100 # each episode takes max_episode_steps / act_steps steps
render_num: 0
@@ -48,7 +48,7 @@ env:
reset_within_step: 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
@@ -23,7 +23,7 @@ horizon_steps: 16
act_steps: 8
use_ddim: True
ddim_steps: 5
ft_denoising_steps: 5
ft_denoising_steps: 0
n_steps: 400 # each episode takes max_episode_steps / act_steps steps
render_num: 0
@@ -63,7 +63,7 @@ shape_meta:
shape: [14]
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