14 lines
355 B
Bash
14 lines
355 B
Bash
|
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
|