Bert-Pytorch-Chinese-TextClassification

Bert-Pytorch-Chinese-TextClassification copied to clipboard

关于test的问题

3

您好，您的代码让我受益匪浅，我也成功完成了训练阶段，我想请教您一下关于test的测试阶段的代码您方便提供我参考学习吗，万分感谢

pbz123

when I run the shell,hava a CUDA error to me. it will runing on a right status if i use cpu.

4

python3 run_classifier_word.py --task_name NEWS --do_train --do_eval --data_dir $GLUE_DIR/ --vocab_file $BERT_BASE_DIR/vocab.txt --bert_config_file $BERT_BASE_DIR/bert_config.json --init_checkpoint $BERT_BASE_DIR/pytorch_model.bin --max_seq_length 256 --train_batch_size 24 --learning_rate 2e-5 --num_train_epochs 50.0 --output_dir ./newsAll_output/ --local_rank 3 04/16/2019 16:33:35 - INFO...

WavesLi

johnnyjana730

Add 2 args: * --from_trained(bool default False) To save and load the model you trained * --is_bi-classification(bool default True) To calculate indexes like precision recall and F1 when doing classification...

807544076