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