#/bin/bash

CUDA_VISIBLE_DEVICES=0 allennlp ltrain training_configs/embert/embert_wide.jsonnet -s storage/models/alfred/moca_36_9_9_9/embert_bs_8_lr_2e-5_segsize_12_split_dot_product_memlen_200_moca_navrecpt_vm --include-package grolp --override "{'dataset_reader': {'mask_token_prob': -0.0, 'mask_object_prob': 0.15}, 'trainer': {'traj_segment_size': 12, 'optimizer': {'lr': 2e-5}}, 'model': {'use_lm_loss': false, 'use_vm_loss': true, 'use_itm_loss': false, 'use_nav_receptacle_loss': true}}";
CUDA_VISIBLE_DEVICES=1 allennlp ltrain training_configs/embert/embert_wide.jsonnet -s storage/models/alfred/moca_36_9_9_9/embert_bs_8_lr_2e-5_segsize_12_split_dot_product_memlen_200_moca_vm --include-package grolp --override "{'dataset_reader': {'mask_token_prob': -0.0, 'mask_object_prob': 0.15}, 'trainer': {'traj_segment_size': 12, 'optimizer': {'lr': 2e-5}}, 'model': {'use_lm_loss': false, 'use_vm_loss': true, 'use_itm_loss': false, 'use_nav_receptacle_loss': false}}";
CUDA_VISIBLE_DEVICES=2 allennlp ltrain training_configs/embert/embert_wide.jsonnet -s storage/models/alfred/moca_36_9_9_9/embert_bs_8_lr_2e-5_segsize_14_split_dot_product_memlen_200_moca_navrecpt_vm --include-package grolp --override "{'dataset_reader': {'mask_token_prob': -0.0, 'mask_object_prob': 0.15}, 'trainer': {'traj_segment_size': 14, 'optimizer': {'lr': 2e-5}}, 'model': {'use_lm_loss': false, 'use_vm_loss': true, 'use_itm_loss': false, 'use_nav_receptacle_loss': true}}";

CUDA_VISIBLE_DEVICES=0 allennlp ltrain training_configs/embert/embert_horizon0.jsonnet -s storage/models/alfred/moca_36_9_9_9-horizon0/embert_bs_8_lr_2e-5_segsize_14_split_dot_product_memlen_200_moca_navrecpt --include-package grolp --override "{'dataset_reader': {'mask_token_prob': -0.0, 'mask_object_prob': 0.0}, 'trainer': {'traj_segment_size': 14, 'optimizer': {'lr': 2e-5}}, 'model': {'use_lm_loss': false, 'use_vm_loss': false, 'use_itm_loss': false, 'use_nav_receptacle_loss': true}}";
CUDA_VISIBLE_DEVICES=1 allennlp ltrain training_configs/embert/embert_horizon0.jsonnet -s storage/models/alfred/moca_36_9_9_9-horizon0/embert_bs_8_lr_2e-5_segsize_12_split_dot_product_memlen_200_moca_navrecpt --include-package grolp --override "{'dataset_reader': {'mask_token_prob': -0.0, 'mask_object_prob': 0.0}, 'trainer': {'traj_segment_size': 12, 'optimizer': {'lr': 2e-5}}, 'model': {'use_lm_loss': false, 'use_vm_loss': false, 'use_itm_loss': false, 'use_nav_receptacle_loss': true}}";
CUDA_VISIBLE_DEVICES=2 allennlp ltrain training_configs/embert/embert_horizon0.jsonnet -s storage/models/alfred/moca_36_9_9_9-horizon0/embert_bs_8_lr_2e-5_segsize_14_split_dot_product_memlen_200_moca_navrecpt_vm --include-package grolp --override "{'dataset_reader': {'mask_token_prob': -0.0, 'mask_object_prob': 0.15}, 'trainer': {'traj_segment_size': 14, 'optimizer': {'lr': 2e-5}}, 'model': {'use_lm_loss': false, 'use_vm_loss': true, 'use_itm_loss': false, 'use_nav_receptacle_loss': true}}";
CUDA_VISIBLE_DEVICES=3 allennlp ltrain training_configs/embert/embert_horizon0.jsonnet -s storage/models/alfred/moca_36_9_9_9-horizon0/embert_bs_8_lr_2e-5_segsize_12_split_dot_product_memlen_200_moca_navrecpt_vm --include-package grolp --override "{'dataset_reader': {'mask_token_prob': -0.0, 'mask_object_prob': 0.15}, 'trainer': {'traj_segment_size': 12, 'optimizer': {'lr': 2e-5}}, 'model': {'use_lm_loss': false, 'use_vm_loss': true, 'use_itm_loss': false, 'use_nav_receptacle_loss': true}}";