13 lines
251 B
Bash
13 lines
251 B
Bash
|
python3 train.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
|