pytorch深度学习中每个epoch运行时间的统计代码

开头部分引入time模块，

import time

然后在每个epoch循环中开始处加入：

since = time.time()

结尾处加入以下代码： 

time_elapsed = time.time() - since
print('Training complete in {:.0f}m {:.0f}s'.format(
    time_elapsed // 60, time_elapsed % 60))