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

use default epoch_start_ema=20 and update_ema_freq=10

Browse Source
This commit is contained in:
allenzren
2024-11-07 10:55:16 -05:00
parent c0921a1fb5
commit 0bdae945e9
82 changed files with 11 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cfg/robomimic/pretrain/lift/pre_diffusion_mlp.yaml
-1
View File
@@ -32,7 +32,6 @@ train:
first_cycle_steps: 10000
warmup_steps: 100
min_lr: 1e-5
epoch_start_ema: 20
save_model_freq: 500
model:
cfg/robomimic/pretrain/lift/pre_diffusion_mlp_img.yaml
-1
View File
@@ -42,7 +42,6 @@ train:
first_cycle_steps: 8000
warmup_steps: 100
min_lr: 1e-5
epoch_start_ema: 20
save_model_freq: 500
model:
cfg/robomimic/pretrain/lift/pre_diffusion_unet.yaml
-1
View File
@@ -32,7 +32,6 @@ train:
first_cycle_steps: 10000
warmup_steps: 100
min_lr: 1e-5
epoch_start_ema: 20
save_model_freq: 500
model:
cfg/robomimic/pretrain/lift/pre_gaussian_mlp.yaml
-1
View File
@@ -31,7 +31,6 @@ train:
first_cycle_steps: 5000
warmup_steps: 100
min_lr: 1e-5
epoch_start_ema: 20
save_model_freq: 500
model:
cfg/robomimic/pretrain/lift/pre_gaussian_mlp_img.yaml
-1
View File
@@ -41,7 +41,6 @@ train:
first_cycle_steps: 3000
warmup_steps: 100
min_lr: 1e-5
epoch_start_ema: 20
save_model_freq: 500
model:
cfg/robomimic/pretrain/lift/pre_gaussian_transformer.yaml
-1
View File
@@ -31,7 +31,6 @@ train:
first_cycle_steps: 5000
warmup_steps: 100
min_lr: 1e-5
epoch_start_ema: 20
save_model_freq: 500
model:
cfg/robomimic/pretrain/lift/pre_gmm_mlp.yaml
-1
View File
@@ -32,7 +32,6 @@ train:
first_cycle_steps: 5000
warmup_steps: 100
min_lr: 1e-5
epoch_start_ema: 20
save_model_freq: 500
model:
cfg/robomimic/pretrain/lift/pre_gmm_transformer.yaml
-1
View File
@@ -32,7 +32,6 @@ train:
first_cycle_steps: 5000
warmup_steps: 100
min_lr: 1e-5
epoch_start_ema: 20
save_model_freq: 500
model: