Я пытаюсь выполнить скрипт BERT run_clasifier.py, используя терминал, как показано ниже:
python run_classifier.py --task_name=cola --do_predict=true --data_dir=<data-dir> --vocab_file=$BERT_BASE_DIR/vocab.txt --bert_config_file=$BERT_BASE_DIR/bert_config.json --init_checkpoint=<output_dir>/model.ckpt-1603 --max_seq_length=512 --output_dir=<output_dir>
Это по умолчанию выполняется на GPU.Вместо этого я хочу выполнить его на процессоре.Есть ли способ сделать это только для одного судебного разбирательства, чем для всех последующих испытаний ??1005*