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

squash commits

Browse Source
This commit is contained in:
allenzren
2024-09-11 21:09:17 -04:00
parent 8ce0aa1485
commit 2ddf63b8f5
200 changed files with 1240 additions and 1186 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cfg/robomimic/pretrain/transport/pre_diffusion_mlp.yaml
-2
View File
@@ -52,9 +52,7 @@ model:
horizon_steps: ${horizon_steps}
obs_dim: ${obs_dim}
action_dim: ${action_dim}
transition_dim: ${transition_dim}
denoising_steps: ${denoising_steps}
cond_steps: ${cond_steps}
device: ${device}
ema:
cfg/robomimic/pretrain/transport/pre_diffusion_mlp_img.yaml
+5 -3
View File
@@ -18,6 +18,7 @@ transition_dim: ${action_dim}
denoising_steps: 100
horizon_steps: 8
cond_steps: 1
img_cond_steps: 1
wandb:
entity: ${oc.env:DPPO_WANDB_ENTITY}
@@ -27,7 +28,7 @@ wandb:
shape_meta:
obs:
rgb:
shape: [3, 96, 96]
shape: [3, 96, 96] # not counting img_cond_steps
state:
shape: [9]
action:
@@ -55,6 +56,7 @@ model:
backbone:
_target_: model.common.vit.VitEncoder
obs_shape: ${shape_meta.obs.rgb.shape}
num_channel: ${eval:'${shape_meta.obs.rgb.shape[0]} * ${img_cond_steps}'} # each image patch is history concatenated
cfg:
patch_size: 8
depth: 1
@@ -68,15 +70,14 @@ model:
time_dim: 32
mlp_dims: [768, 768, 768]
residual_style: True
img_cond_steps: ${img_cond_steps}
cond_dim: ${eval:'${obs_dim} * ${cond_steps}'}
horizon_steps: ${horizon_steps}
transition_dim: ${transition_dim}
horizon_steps: ${horizon_steps}
obs_dim: ${obs_dim}
action_dim: ${action_dim}
transition_dim: ${transition_dim}
denoising_steps: ${denoising_steps}
cond_steps: ${cond_steps}
device: ${device}
ema:
@@ -89,4 +90,5 @@ train_dataset:
horizon_steps: ${horizon_steps}
max_n_episodes: 100
cond_steps: ${cond_steps}
img_cond_steps: ${img_cond_steps}
device: ${device}
cfg/robomimic/pretrain/transport/pre_diffusion_unet.yaml
-2
View File
@@ -55,9 +55,7 @@ model:
horizon_steps: ${horizon_steps}
obs_dim: ${obs_dim}
action_dim: ${action_dim}
transition_dim: ${transition_dim}
denoising_steps: ${denoising_steps}
cond_steps: ${cond_steps}
device: ${device}
ema:
cfg/robomimic/pretrain/transport/pre_gaussian_mlp_img.yaml
+5 -1
View File
@@ -17,6 +17,7 @@ action_dim: 14
transition_dim: ${action_dim}
horizon_steps: 8
cond_steps: 1
img_cond_steps: 1
wandb:
entity: ${oc.env:DPPO_WANDB_ENTITY}
@@ -26,7 +27,7 @@ wandb:
shape_meta:
obs:
rgb:
shape: [3, 96, 96]
shape: [3, 96, 96] # not counting img_cond_steps
state:
shape: [9]
action:
@@ -52,6 +53,7 @@ model:
backbone:
_target_: model.common.vit.VitEncoder
obs_shape: ${shape_meta.obs.rgb.shape}
num_channel: ${eval:'${shape_meta.obs.rgb.shape[0]} * ${img_cond_steps}'} # each image patch is history concatenated
cfg:
patch_size: 8
depth: 1
@@ -65,6 +67,7 @@ model:
mlp_dims: [768, 768, 768]
residual_style: True
fixed_std: 0.1
img_cond_steps: ${img_cond_steps}
cond_dim: ${eval:'${obs_dim} * ${cond_steps}'}
horizon_steps: ${horizon_steps}
transition_dim: ${transition_dim}
@@ -81,4 +84,5 @@ train_dataset:
horizon_steps: ${horizon_steps}
max_n_episodes: 100
cond_steps: ${cond_steps}
img_cond_steps: ${img_cond_steps}
device: ${device}