欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

一些常用的函数技巧

程序员文章站 2022-03-11 08:33:28
...

1. 计算模型训练时间

import time

start = time.time()
# 输出训练所耗总时间
time_all = time.time() - start
print('Training complete in {:.2f}m | {:.2f}s'.format(time_all // 60, time_all % 60))

相关标签: pytorch python