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