PyTorch Lightning Examples

描述

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

安装

Windows 11

conda create -n ai-code-examples python=3.8
conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit==11.3 -c pytorch
conda install -c conda-forge torchmetrics
conda install -c conda-forge onnxruntime
pip install lightning

引用

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