2023年2月17日
测试pytorch时间
摘要： torch.cuda.synchronize() start = time.time() result = model(return_loss=False, rescale=True, **data) torch.cuda.synchronize() end = time.time() print( 阅读全文
posted @ 2023-02-17 18:05 无左无右 阅读(24) 评论(0) 推荐(0) 编辑