1a1308d62f
add fixes for reproducibility only try to start env if it is closed revision fix normalization and data type Improve README Improve README Tests are passing, Eval pretrained model works, Add gif Update gif Update gif Update gif Update gif Update README Update README update minor Update README.md Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com> Update README.md Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com> Address suggestions Update thumbnail + stats Update thumbnail + stats Update README.md Co-authored-by: Alexander Soare <alexander.soare159@gmail.com> Add more comments Add test_examples.py
43 lines
814 B
YAML
43 lines
814 B
YAML
defaults:
|
|
- _self_
|
|
- env: pusht
|
|
- policy: diffusion
|
|
|
|
hydra:
|
|
run:
|
|
dir: outputs/train/${now:%Y-%m-%d}/${now:%H-%M-%S}_${env.name}_${policy.name}_${hydra.job.name}
|
|
job:
|
|
name: default
|
|
|
|
seed: 1337
|
|
# batch size for TorchRL SerialEnv. Each underlying env will get the seed = seed + env_index
|
|
# NOTE: only diffusion policy supports rollout_batch_size > 1
|
|
rollout_batch_size: 1
|
|
device: cuda # cpu
|
|
prefetch: 4
|
|
eval_freq: ???
|
|
save_freq: ???
|
|
eval_episodes: ???
|
|
save_video: false
|
|
save_model: false
|
|
save_buffer: false
|
|
train_steps: ???
|
|
fps: ???
|
|
|
|
offline_prioritized_sampler: true
|
|
|
|
dataset_id: ???
|
|
|
|
n_action_steps: ???
|
|
n_obs_steps: ???
|
|
env: ???
|
|
|
|
policy: ???
|
|
|
|
wandb:
|
|
enable: true
|
|
# Set to true to disable saving an artifact despite save_model == True
|
|
disable_artifact: false
|
|
project: lerobot
|
|
notes: ""
|