pytorch 正确的测试时间的代码 torch.cuda.synchronize()
1
start = time.time()
result = model(input)
end = time.time()
2
torch.cuda.synchronize()
start = time.time()
result = model(input)
torch.cuda.synchronize()
end = time.time()
一共上述两种测试时间的方式,正确的方式是第二种,...
转载
2020-02-25 21:10:40 ·
2458 阅读 ·
0 评论