换脸拥抱变压器run_clm.py提早停止

问题描述

我正在运行run_clm.py，以按照language_modeling示例对gpt-2进行优化，使其形成了拥抱面库：

!python run_clm.py \
    --model_name_or_path gpt2 \
    --train_file train.txt \
    --validation_file test.txt \
    --do_train \
    --do_eval \
    --output_dir /tmp/test-clm

这是输出，进程似乎已经开始，但是出现了 ^ C 来停止进程：

The following columns in the training set don't have a corresponding argument in `GPT2LMHeadModel.forward` and have been ignored: .
The following columns in the evaluation set don't have a corresponding argument in `GPT2LMHeadModel.forward` and have been ignored: .
***** Running training *****
  Num examples = 2318
  Num Epochs = 3
  Instantaneous batch size per device = 8
  Total train batch size (w. parallel,distributed & accumulation) = 8
  Gradient Accumulation steps = 1
  Total optimization steps = 870
  0% 0/870 [00:00<?,?it/s]^C

这是我的环境信息：

变形金刚版本：3.4.0
平台：Linux-4.19.112 + -x86_64-with-ubuntu-18.04-bionic
Python版本：3.6.9
Tensorflow版本：1.14
在脚本中使用GPU ？：是

提前停止的可能触发因素是什么？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

gpt-2 huggingface-transformers