torch.autograd.profiler.profile()

简单来说,这只是一个辅助函数,帮你弄清楚底层计算机运行情况而已。

参考自:Pytorch剖析器及Pytorch模型的逐层分析 - 那抹阳光1994 - 博客园 (cnblogs.com)

你可能感兴趣的:(CV笔记,深度学习)