2025年2月9日
PyTorch Profiler 性能优化示例：定位 TorchMetrics 收集瓶颈，提高 GPU 利用率
摘要： 指标收集是每个机器学习项目不可或缺的组成部分，它使我们能够跟踪模型性能并监控训练进度。理想情况下，我们希望在不给训练过程带来额外开销的前提下收集和计算指标。与训练循环的其他部分一样，低效的指标计算可能会引入不必要的开销，延长训练步骤的耗时，并增加训练成本。 本文是将聚焦于指标收集，演示指标收集的一种 阅读全文
posted @ 2025-02-09 11:28 deephub 阅读(16) 评论(0) 推荐(0) 编辑