tim = 0 start = torch.cuda.Event(enable_timing=True) #the times end = torch.cuda.Event(enable_timing=True) start.record() out = net(image_input) end.record() torch.cuda.synchronize() tim = start.elapsed_time(end)
08-01
1225
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
01-16
1274
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
10-19
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交