File size: 311 Bytes
5381499 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
python3 main.py \
--batch_size 2 \
--num_workers 2 \
--classifier_lr 1e-4 \
--wav2vec2_lr 1e-5 \
--max_epochs 10 \
--accelerator cpu \
--weight_decay 0.001 \
--warmup_steps 0.1 \
--constant_steps 0.4 \
--scheduler_factor 0.001 \
--data_dir data \
--ckpt_dir ckpt
|