BASE_DIR="/mnt/sun-pcp01/bmattinson/route_frame/spose/v9/v9_0"

EXPERIMENT_NAME="v9_0_0"
RESUME_EXPERIMENT_ID="pos_sub_1_0"
SOLVER_FN="$BASE_DIR/$EXPERIMENT_NAME/solver.pbtxt"
TRAIN_FN="$BASE_DIR/$EXPERIMENT_NAME/pos_train.pbtxt"
INFERENCE_FN="$BASE_DIR/$EXPERIMENT_NAME/pos_model_unmerged.pbtxt"
MODEL_OVERRIDE_FN=""
DATAMATRIX_DIR="/mnt/sun-pcp01/bmattinson/route_frame/spose/v9/features"
LMDB_DIR="/mnt/sun-pcs01/bmattinson/route_frame/spose/lmdbs"
MODEL_DIR="$BASE_DIR"
EVAL_APP=""
STAGES="multiblob,train,trt"
SUBSAMPLE_PCT="1.0"

BATCH_SIZE="128"
BB_STRAIGHT="10"
BB_LEFT="10"
BB_LEFT_WITH_ONCOMING="1"
BB_RIGHT="10"

BLOB_CREATOR="JunctionPosePositionOutputBlobCreator"

BATCH_BALANCE_TEST="1"

pipedream/tools/run bazel --config=cuda -c opt --gpus 1 --mailto=bmattinson@zoox.com //experimental/bmattinson/train_automator:train_automator -- --experiment_group_name=$EXPERIMENT_NAME --solver_fn=$SOLVER_FN --train_fn=$TRAIN_FN --inference_fn=$INFERENCE_FN --model_override_fn=$MODEL_OVERRIDE_FN --datamatrix_dir=$DATAMATRIX_DIR --lmdb_dir=$LMDB_DIR --model_dir=$MODEL_DIR --eval_app=$EVAL_APP --stages=$STAGES --subsample_pct=$SUBSAMPLE_PCT --resume_experiment_id=$RESUME_EXPERIMENT_ID --batch_size=$BATCH_SIZE --bb_straight=$BB_STRAIGHT --bb_left=$BB_LEFT --bb_left_with_oncoming=$BB_LEFT_WITH_ONCOMING --bb_right=$BB_RIGHT --blob_creator=$BLOB_CREATOR --batch_balance_test=$BATCH_BALANCE_TEST


#brun --config=cuda //experimental/bmattinson/train_automator:train_automator -- --experiment_group_name=$EXPERIMENT_NAME --solver_fn=$SOLVER_FN --train_fn=$TRAIN_FN --inference_fn=$INFERENCE_FN --model_override_fn=$MODEL_OVERRIDE_FN --datamatrix_dir=$DATAMATRIX_DIR --lmdb_dir=$LMDB_DIR --model_dir=$MODEL_DIR --eval_app=$EVAL_APP --stages=$STAGES --subsample_pct=$SUBSAMPLE_PCT --resume_experiment_id=$RESUME_EXPERIMENT_ID --batch_size=$BATCH_SIZE --bb_straight=$BB_STRAIGHT --bb_left=$BB_LEFT --bb_left_with_oncoming=$BB_LEFT_WITH_ONCOMING --bb_right=$BB_RIGHT --blob_creator=$BLOB_CREATOR --batch_balance_test=$BATCH_BALANCE_TEST

