Deep learning framework (PyTorch Lightning). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard), distributed training (DDP, FSDP, DeepSpeed), for scalable neural network training.
深度学习框架(PyTorch Lightning)。将PyTorch代码组织为LightningModules,配置训练器以支持多GPU/TPU,实现数据管道、回调函数、日志记录(W&B、TensorBoard)以及分布式训练(DDP、FSDP、DeepSpeed),以实现可扩展的神经网络训练。
直接复制以下提示词,发送给你的 AI 助手即可完成安装。
点击右上角 下载SKILL 按钮