在transformers中output_dir怎么设置

Question

跑通transformers时无法写入output_dir

在pytorch环境中运行的

the following arguments are required: --output_dir

在pycharm中用运行时插入参数的方法不行

会提示 Some specified arguments are not used by the HfArgumentParser: ['D:\transformers']

export GLUE_DIR=/path/to/glue
export TASK_NAME=MRPC

python ./examples/text-classification/run_glue.py \
    --model_name_or_path bert-base-uncased \
    --task_name $TASK_NAME \
    --do_train \
    --do_eval \
    --data_dir $GLUE_DIR/$TASK_NAME \
    --max_seq_length 128 \
    --per_device_eval_batch_size=8   \
    --per_device_train_batch_size=8   \
    --learning_rate 2e-5 \
    --num_train_epochs 3.0 \
    --output_dir /tmp/$TASK_NAME/D:\transformers

只想跑通transformers

Answer 1

猜想你跑的脚本使用了argparse类，argparse中定义了变量--output_dir
你需要给output_dir赋值，具体的解决方案还是把脚本发一下吧。