计算一下pytorch中Resnet34模型前传一次所需要的时间
来源:互联网 发布:n9软件下载 编辑:程序博客网 时间:2024/05/16 01:57
import pdbimport timeimport torchimport torchvision.models as modelsfrom torch.autograd import Variableclass Timer(object): """A simple timer.""" def __init__(self): self.total_time = 0. self.calls = 0 self.start_time = 0. self.diff = 0. self.average_time = 0. def tic(self): # using time.time instead of time.clock because time time.clock # does not normalize for multithreading self.start_time = time.time() def toc(self, average=True): self.diff = time.time() - self.start_time self.total_time += self.diff self.calls += 1 self.average_time = self.total_time / self.calls if average: return self.average_time else: return self.diffGPUID = 1resnet34 = models.resnet34(pretrained=True)resnet34.cuda(GPUID)x = torch.rand(1,3,400,400)x = Variable(x.cuda(GPUID))# preheaty = resnet34(x)timer = Timer()timer.tic()for i in xrange(100): y = resnet34(x)timer.toc()print ('Do once forward need {:.3f}ms ').format(timer.total_time*1000/100.0)
阅读全文
0 0
- 计算一下pytorch中Resnet34模型前传一次所需要的时间
- 计算Java程序运行所需要的时间
- pytorch 模型的加载
- 取得前一次MySQL操作所影响的记录行数
- 取得前一次MySQL操作所影响的记录行数
- 得前一次MySQL操作所影响的记录行数
- 取得前一次MySQL操作所影响的记录行数
- 注册域名前需要注意的,备案所需材料
- 计算几天前,几小时前,几分钟前,几周前,几个月前,几年前的时间
- mysql的sql语句中直接计算时间 查询昨天 一周前 一月前 一年前的数据的方法
- mysql的sql语句中直接计算时间 查询昨天 一周前 一月前 一年前的数据的方法
- java中,如何计算一个方法所耗费的时间秒数?
- QlikView显示所选时间前一年的数据
- usbhid中所需要的函数
- OC中所需要用到的重点
- 如何计算产生SPWM所需要的占空比
- 如何计算产生SPWM所需要的占空比
- iOS 计算UILable显示文字所需要的行数
- 统计学习方法(第三章)k近邻算法
- 初级web api的设计(二)——防重放攻击
- 神经网络 tensorflow教程 2.2 下载MNIST 数据集 (自动版)
- 统计学习方法总结、应用对比
- 《Android学习笔记》利用Application操作全局变量
- 计算一下pytorch中Resnet34模型前传一次所需要的时间
- tf.reshape()的用法与理解
- 写出程序的输出结果
- Java正则表达式
- Warning:The `android.dexOptions.incremental` property is deprecated and it has no effect on the buil
- 安装和配置git
- 自定义控件实现banner轮播
- 最大流的Ford-Fulkerson 标号法
- 学习记录2