Link-Li / pytorch-lightning-learn

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch-lightning-learn

简介

这部分是知乎上PyTorch Lightning入门教程的代码,代码分为三部分,分别对应知乎中的三个例子。

这里使用的python包的版本参考文件requirements.txt,这里采用anaconda导出的环境,安装命令如下:

conda install --yes --file requirements.txt

其中比较重要的几个环境的版本如下:

python3.7
pytorch=1.7.1
pytorch-lightning=1.5.10
transformers =4.21.1

运行方式

针对ResNet-50运行代码:

  • 如果无法自动下载ResNet-50的模型和CIFAR10的数据集,建议使用代理之后再运行代码
cd 1-Resnet-50
python main.py

针对BERT-base运行代码:

cd 2-Bert-base
python main.py

针对T5-base运行代码:

cd 3-T5-base
python main.py
  • 预测模型的生成结果:
cd 3-T5-base
python predict.py --limit_predict_batches 1.0

这里参考了如下的一些代码和内容:

https://www.pytorchlightning.ai/

https://github.com/renmada/t5-pegasus-pytorch

情感分析数据集来自:https://zhuanlan.zhihu.com/p/80029681

文本摘要数据集来自:https://zhuanlan.zhihu.com/p/341398288

About


Languages

Language:Python 100.0%