File size: 613 Bytes
f96995c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
CUDA_VISIBLE_DEVICES=0 python experiments/train/train_eval.py \
'gpus=[0]' \
overwrite=False \
resume=True \
debug=True \
train.name=sloth/train \
train.source_dataset_name=data/sloth_merged/sub_episodes_v \
train.dataset_name=sloth/dataset \
train.dataset_load_skip_frame=3 \
train.dataset_skip_frame=1 \
train.num_workers=8 \
train.batch_size=32 \
train.training_start_episode=0 \
train.training_end_episode=113 \
train.eval_start_episode=113 \
train.eval_end_episode=133 \
train.num_iterations=100000 \
train.iteration_eval_interval=1000 \
train.iteration_log_interval=10 \
train.iteration_save_interval=1000
|