File size: 709 Bytes
f96995c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
CUDA_VISIBLE_DEVICES=0 python experiments/train/train_eval.py \
'gpus=[0]' \
overwrite=False \
resume=True \
debug=True \
train.name=box/train \
train.source_dataset_name=data/box_merged/sub_episodes_v \
train.dataset_name=box/dataset \
sim.num_grippers=2 \
sim.gripper_points=True \
sim.gripper_forcing=False \
model.material.radius=0.0 \
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=303 \
train.eval_start_episode=303 \
train.eval_end_episode=323 \
train.num_iterations=100000 \
train.iteration_eval_interval=1000 \
train.iteration_log_interval=10 \
train.iteration_save_interval=1000
|