技术栈
lightning
Sylvan Ding
24 天前
人工智能
·
pytorch
·
python
·
lightning
PyTorch Lightning实战 - 训练 MNIST 数据集
利用 PyTorch Lightning 训练 MNIST 数据。验证梯度范数、学习率、优化器对训练的影响。
牛andmore牛
2 年前
fabric
·
lightning
·
pytorch多gpu
·
pytorch混合精度
2、基于pytorch lightning的fabric实现pytorch的多GPU训练和混合精度功能
输出结果:以上fabric对应lightning2.1版本,该工具还在开发中,后期会有其他功能。