dodox/dppo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update eval configs

Browse Source
This commit is contained in:
allenzren
2025-02-04 11:51:47 -05:00
committed by Allen Z. Ren
parent ace2bbdab9
commit 169a16dda7
27 changed files with 81 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cfg/robomimic/eval/square/eval_diffusion_mlp.yaml
+3 -1
View File
@@ -20,6 +20,7 @@ denoising_steps: 20
cond_steps: 1
horizon_steps: 4
act_steps: 4
ft_denoising_steps: 10
n_steps: 400 # each episode takes max_episode_steps / act_steps steps
render_num: 0
@@ -44,7 +45,8 @@ env:
reset_within_step: True
model:
_target_: model.diffusion.diffusion.DiffusionModel
_target_: model.diffusion.diffusion_eval_ft.DiffusionEvalFT
ft_denoising_steps: ${ft_denoising_steps}
predict_epsilon: True
denoised_clip_value: 1.0
randn_clip_value: 3
cfg/robomimic/eval/square/eval_diffusion_mlp_img.yaml
+3 -1
View File
@@ -23,6 +23,7 @@ horizon_steps: 4
act_steps: 4
use_ddim: True
ddim_steps: 5
ft_denoising_steps: 5
n_steps: 400 # each episode takes max_episode_steps / act_steps steps
render_num: 0
@@ -58,7 +59,8 @@ shape_meta:
shape: [7]
model:
_target_: model.diffusion.diffusion.DiffusionModel
_target_: model.diffusion.diffusion_eval_ft.DiffusionEvalFT
ft_denoising_steps: ${ft_denoising_steps}
predict_epsilon: True
denoised_clip_value: 1.0
randn_clip_value: 3
cfg/robomimic/eval/square/eval_diffusion_unet.yaml
+3 -1
View File
@@ -20,6 +20,7 @@ denoising_steps: 20
cond_steps: 1
horizon_steps: 4
act_steps: 4
ft_denoising_steps: 10
n_steps: 100 # each episode takes max_episode_steps / act_steps steps
render_num: 0
@@ -44,7 +45,8 @@ env:
reset_within_step: True
model:
_target_: model.diffusion.diffusion.DiffusionModel
_target_: model.diffusion.diffusion_eval_ft.DiffusionEvalFT
ft_denoising_steps: ${ft_denoising_steps}
predict_epsilon: True
denoised_clip_value: 1.0
randn_clip_value: 3
cfg/robomimic/eval/square/eval_diffusion_unet_img.yaml
+3 -1
View File
@@ -23,6 +23,7 @@ horizon_steps: 4
act_steps: 4
use_ddim: True
ddim_steps: 5
ft_denoising_steps: 5
n_steps: 400 # each episode takes max_episode_steps / act_steps steps
render_num: 0
@@ -58,7 +59,8 @@ shape_meta:
shape: [7]
model:
_target_: model.diffusion.diffusion.DiffusionModel
_target_: model.diffusion.diffusion_eval_ft.DiffusionEvalFT
ft_denoising_steps: ${ft_denoising_steps}
predict_epsilon: True
denoised_clip_value: 1.0
randn_clip_value: 3