使用PyTorch Profiler进行模型性能分析,改善并加速PyTorch训练

如果所有机器学习工程师都想要一样东西,那就是更快的模型训练——也许在良好的测试指标之后

加速机器学习模型训练是所有机器学习工程师想要的一件事。更快的训练等于更快的实验,更快的产品迭代,还有最重要的一点需要更少的资源,也就是更省钱。

熟悉PyTorch Profiler

然后就可以启动tensorboard查看分析轨迹。如果这一步有问题,请查看是否安装了torch-tb-profiler。

https://avoid.overfit.cn/post/95f7fa956805466db713e797d9d62e67

posted @ 2024-06-12 12:06  deephub  阅读(8)  评论(0编辑  收藏  举报