随笔分类 -  Pytorch

摘要:训练时间统计 1 引入time模块 import time 2 在epoch循环开头处加入 begin = time.time() 3 在epoch循环末尾处加入 time_cost = time.time() - begin print('Training complete in {:.0f}m 阅读全文
posted @ 2021-10-03 17:55 SethDeng 阅读(95) 评论(0) 推荐(0)