# PyTorch Lightning Examples

描述

这是一个 pytorch lightning 的样例仓库，里面有帮助文件中的样例，也有工作需要使用的模型。

如何运行

首先，安装依赖

# clone project   
git clone https://github.com/zhuyuanxiang/pytorch_lighting_example

# install project   
cd pytorch_lighting_example 
pip install -e .   
pip install -r requirements.txt

然后，浏览任意文件，并且运行它。

# 项目文件夹
cd project

# 运行模型
python train_classifier.py    

导入

项目作为包的设置可以轻松地导入任意的文件，如下：

from project.torch_datasets.mnist import mnist
from project.lit_classifier_main import LitClassifier
from pytorch_lightning import Trainer

# model
model = LitClassifier()

# datasets
train, val, test = mnist()

# train
trainer = Trainer()
trainer.fit(model, train, val)

# test using the best model!
trainer.test(test_dataloaders=test)

Citation

@article{Zhuyx,
  title={Pytorch-Lightning Examples},
  author={zhuyuanxiang},
  journal={China},
  year={2022}
}