技术栈
pytorch混合精度
牛andmore牛
1 年前
fabric
·
lightning
·
pytorch多gpu
·
pytorch混合精度
2、基于pytorch lightning的fabric实现pytorch的多GPU训练和混合精度功能
输出结果:以上fabric对应lightning2.1版本,该工具还在开发中,后期会有其他功能。