Skip to content

Latest commit

 

History

History
6 lines (6 loc) · 619 Bytes

Training.md

File metadata and controls

6 lines (6 loc) · 619 Bytes

Training

For training a Visuomotor Policy, please use the following commands.

python3 scripts/train.py --config=PATH_TO_CONFIG --exp=EXPERIMENT_NAME --env=ENV_TYPE --subtask=SUBTASK_INDEX --data=PATH_TO_DATASET

The configuration at ./config/trill.json would be used for training as the default unless you specify PATH_TO_CONFIG. Trained files would be saved in ./save/EXPERIMENT_NAME/{ENV_TYPE}_{SUBTASK_INDEX}. You need to create or download (link) an hdf5-format dataset file and specify the path to the dataset file as PATH_TO_CONFIG.