在transformers中output_dir怎么设置

跑通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

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


parse.add_argument('--output_dir',type = str,default ="这里填上路径",required=True)

找到代码中的output这一行,里面像我这样添加上default参数,然后就可以在pycharm中直接运行了