-
Notifications
You must be signed in to change notification settings - Fork 28
/
training_run_notes_language.txt
20 lines (13 loc) · 3.93 KB
/
training_run_notes_language.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
python3 main.py --is_sim --obj_mesh_dir objects/blocks --num_obj 8 --push_rewards --experience_replay --explore_rate_decay --trial_reward --common_sense --place --future_reward_discount 0.65 --max_train_actions 20000 --random_actions --depth_channels_history --static_language_mask --language_model_config ~/Downloads/gr_model/config.yaml --language_model_weights ~/Downloads/gr_model/best.th --is_testing
# Transformer stacking multiple timestep
export CUDA_VISIBLE_DEVICES="0" && python3 main.py --is_sim --is_testing --obj_mesh_dir objects/blocks --num_obj 8 --common_sense --place --depth_channels_history --static_language_mask --language_model_config models/gr_train_augment_only_depthmaps_fixed/config.yaml --language_model_weights models/gr_train_augment_only_depthmaps_fixed/best.th --snapshot_file models/2020-06-05-18-28-46_Real-Stack-SPOT-Trial-Reward-Masked-Testing/models/snapshot.reinforcement_action_efficiency_best_value.pth --goal_num_obj=4 --end_on_incorrect_order
# Transformer stacking single timestep
export CUDA_VISIBLE_DEVICES="0" && python3 main.py --is_sim --is_testing --obj_mesh_dir objects/blocks --num_obj 8 --common_sense --place --static_language_mask --language_model_config models/gr_train_augment_only_depthmaps_fixed/config.yaml --language_model_weights models/gr_train_augment_only_depthmaps_fixed/best.th --snapshot_file models/good-robot-best-single-timestep-models/2020-06-07-21-42-16_Sim-Stack-SPOT-Trial-Reward-Masked-Training-best-model-good-robot-paper/snapshot.reinforcement_trial_success_rate_best_index.pth --goal_num_obj=4 --end_on_incorrect_order
# Unet stacking single timestep
export CUDA_VISIBLE_DEVICES="0" && python3 main.py --is_sim --is_testing --obj_mesh_dir objects/blocks --num_obj 8 --common_sense --place --static_language_mask --language_model_config models/gr_unet_stacks/config.yaml --language_model_weights models/gr_unet_stacks/best.th --snapshot_file models/good-robot-best-single-timestep-models/2020-06-07-21-42-16_Sim-Stack-SPOT-Trial-Reward-Masked-Training-best-model-good-robot-paper/snapshot.reinforcement_trial_success_rate_best_index.pth --end_on_incorrect_order --goal_num_obj=4 --timeout 100000 --place
# Random stacking
export CUDA_VISIBLE_DEVICES="0" && python3 main.py --is_sim --is_testing --obj_mesh_dir objects/blocks --num_obj 8 --common_sense --place --depth_channels_history --static_language_mask --language_model_config models/gr_train_augment_only_depthmaps_fixed/config.yaml --language_model_weights models/gr_train_augment_only_depthmaps_fixed/best.th --snapshot_file models/2020-06-05-18-28-46_Real-Stack-SPOT-Trial-Reward-Masked-Testing/models/snapshot.reinforcement_action_efficiency_best_value.pth --goal_num_obj=4 --end_on_incorrect_order --baseline_language_mask
# Transformer row making single timestep
export CUDA_VISIBLE_DEVICES="0" && python3 main.py --is_sim --is_testing --obj_mesh_dir objects/blocks --num_obj 4 --static_language_mask --language_model_config models/gr_rows_augment_only_depthmaps_fixed/config.yaml --language_model_weights models/gr_rows_augment_only_depthmaps_fixed/best.th --snapshot_file models/good-robot-best-single-timestep-models/2020-06-03-12-05-28_Sim-Rows-Two-Step-Reward-Masked-Training-best-model/snapshot.reinforcement_trial_success_rate_best_index.pth --end_on_incorrect_order --check_row --timeout 100000 --place --separation_threshold 0.08 --distance_threshold 0.04
# Unet row making single timestep
export CUDA_VISIBLE_DEVICES="0" && python3 main.py --is_sim --is_testing --obj_mesh_dir objects/blocks --num_obj 4 --static_language_mask --language_model_config models/gr_unet_rows/config.yaml --language_model_weights models/gr_unet_rows/best.th --snapshot_file models/good-robot-best-single-timestep-models/2020-06-03-12-05-28_Sim-Rows-Two-Step-Reward-Masked-Training-best-model/snapshot.reinforcement_trial_success_rate_best_index.pth --end_on_incorrect_order --check_row --timeout 100000 --place --separation_threshold 0.08 --distance_threshold 0.04