main.sh 528 B

12345678910111213141516171819202122232425
  1. !python main.py \
  2. --bert_dir="../model_hub/chinese-roberta-wwm-ext/" \
  3. --data_dir="../data/" \
  4. --log_dir="./logs/" \
  5. --output_dir="./checkpoints/" \
  6. --num_tags=5 \
  7. --seed=123 \
  8. --gpu_ids="-1" \
  9. --max_seq_len=300 \
  10. --lr=3e-5 \
  11. --crf_lr=3e-2 \
  12. --other_lr=3e-4 \
  13. --train_batch_size=32 \
  14. --train_epochs=1 \
  15. --eval_batch_size=32 \
  16. --max_grad_norm=1 \
  17. --warmup_proportion=0.1 \
  18. --adam_epsilon=1e-8 \
  19. --weight_decay=0.01 \
  20. --lstm_hidden=128 \
  21. --num_layers=1 \
  22. --use_lstm='True' \
  23. --use_crf='True' \
  24. --dropout_prob=0.3 \
  25. --dropout=0.3 \