first commit

watch_and_help/scripts/test_watch_strategy_full.sh

@@ -0,0 +1,13 @@
+python3 watch_strategy_full/predicate-train-strategy.py \
+--testset test_task \
+--gpu_id 0 \
+--batch_size 32 \
+--demo_hidden 512 \
+--model_type lstmlast \
+--dropout 0 \
+--inputtype actioninput \
+--inference 2 \
+--single 1 \
+--resume '' \
+--loss_type ce \
+--checkpoint checkpoints/train_strategy_full/lstmlast_cross_entropy_bs_32_iter_2000_train_task_prob

watch_and_help/scripts/train_watch_strategy_full.sh

@@ -0,0 +1,13 @@
+python3 watch_strategy_full/predicate-train-strategy.py \
+--gpu_id 0 \
+--model_lr_rate 3e-4 \
+--batch_size 32 \
+--demo_hidden 512 \
+--model_type lstmlast \
+--inputtype actioninput \
+--dropout 0 \
+--single 1 \
+--resume '' \
+--checkpoint checkpoints/train_strategy_full/lstmlast \
+--train_iters 2000 \
+--loss_type ce\