InferringIntention/keyboard_and_mouse/test.sh

13 lines
250 B
Bash
Raw Permalink Normal View History

2024-03-24 23:42:27 +01:00
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