我在 Google Colab 上使用 Pytorch 的 Trainer 类时遇到问题,因为它要求 accelarate>=0.21.0,即使我已经更新了所有要求,还有其他选择吗?
“将
Trainer
与 PyTorch
一起使用需要 accelerate>=0.21.0
:请运行 pip install transformers[torch]
或 pip install accelerate -U
”。
我已经尝试过上述两种建议,但都没有奏效。我不知道我做错了什么或如何继续?
我在 NLP 课程的 Colab 笔记本中遇到了同样的错误:第 3 课 - 使用 Trainer API 微调模型。为了解决这个问题,您可以:
- 在单元格中运行 pip install Accelerate -U
- 在顶部菜单中单击“运行时”→“重新启动运行时”
- 请勿重新运行任何带有 !pip install 的单元
- 重新运行所有其他代码单元,您应该可以开始了!
顺便说一句,请务必通过单击顶部菜单中的编辑 → 笔记本设置 → GPU 类型来打开此笔记本的 GPU。这是让我感动的下一件事:)
来源:https://discuss.huggingface.co/t/trainingargument-does-not-work-on-colab/43372/6