可插拔训练加速trick-Scaling PyTorch Model Training With Minimal Code Changes
依赖:pipinstalllightning插拔改动:fromlightning.fabricimportFabric#...#实例化fabric=Fabric(accelerator='cuda')#混精度用这个,加速明显#fabric=Fabric(accelerator="cuda",precision="bf16-mixed")fabric.launch()#...#插拔接入model,o