47 lines
1.3 KiB
YAML
47 lines
1.3 KiB
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
|
|
|
|
device: cuda # cpu
|
|
# `use_amp` determines whether to use Automatic Mixed Precision (AMP) for training and evaluation. With AMP,
|
|
# automatic gradient scaling is used.
|
|
use_amp: false
|
|
# `seed` is used for training (eg: model initialization, dataset shuffling)
|
|
# AND for the evaluation environments.
|
|
seed: ???
|
|
dataset_repo_id: lerobot/pusht
|
|
|
|
training:
|
|
offline_steps: ???
|
|
# NOTE: `online_steps` is not implemented yet. It's here as a placeholder.
|
|
online_steps: ???
|
|
online_steps_between_rollouts: ???
|
|
online_sampling_ratio: 0.5
|
|
# `online_env_seed` is used for environments for online training data rollouts.
|
|
online_env_seed: ???
|
|
eval_freq: ???
|
|
save_freq: ???
|
|
log_freq: 250
|
|
save_model: true
|
|
|
|
eval:
|
|
n_episodes: 1
|
|
# `batch_size` specifies the number of environments to use in a gym.vector.VectorEnv.
|
|
batch_size: 1
|
|
# `use_async_envs` specifies whether to use asynchronous environments (multiprocessing).
|
|
use_async_envs: false
|
|
|
|
wandb:
|
|
enable: false
|
|
# Set to true to disable saving an artifact despite save_model == True
|
|
disable_artifact: false
|
|
project: lerobot
|
|
notes: ""
|