13 lines
316 B
Bash
13 lines
316 B
Bash
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\
|