13 lines
250 B
Bash
13 lines
250 B
Bash
|
python3 test.py \
|
||
|
--resume False \
|
||
|
--batch_size 8 \
|
||
|
--lr 1e-4 \
|
||
|
--model_type lstmlast \
|
||
|
--epochs 100 \
|
||
|
--demo_hidden 128 \
|
||
|
--hidden_size 128 \
|
||
|
--dropout 0.5 \
|
||
|
--dataset_path dataset/strategy_dataset/ \
|
||
|
--checkpoint checkpoints/ \
|
||
|
--weight_decay 1e-4
|