File size: 622 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=paperbag/train \
train.source_dataset_name=data/paperbag_merged/sub_episodes_v \
train.dataset_name=paperbag/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=200 \
train.eval_start_episode=200 \
train.eval_end_episode=220 \
train.num_iterations=100000 \
train.iteration_eval_interval=1000 \
train.iteration_log_interval=10 \
train.iteration_save_interval=1000
|