微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决换脸拥抱变压器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 举报,一经查实,本站将立刻删除。