PyTorch ResNet 测试

原文：

https://blog.csdn.net/shanglianlm/article/details/86376627

resnet50 1070 gpu上需要13-15ms

resnet18 p100 512*512 gpu上需要6ms

resnet34:

p100 512*512 gpu 10ms

1070 1 416*416 gpu 11ms

1070 8 416*416 gpu 46ms

resnet50 p100 512*512 gpu上需要20ms

ShuffleNetV2 p100 512*512 gpu上需要 20ms

import timeimport torch.nn as nn
from torch.nn.parameter import Parameter
import torch
def conv3x3(in_planes, out_planes, stride=1):"""3x3 convolution with padding"""return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride,padding=1, bias=False)def conv1x1(in_planes, out_planes, stride=1):"""1x1 convolution"""return nn.Conv2d(in_planes, out_planes, kernel_size=1, stride=stride,

PyTorch ResNet 测试相关推荐

PyTorch ResNet 实现图片分类
PyTorch ResNet 实现图片分类建党 100 年 Resnet 深度网络退化代码实现残差块超参数 ResNet 18 网络获取数据训练测试完整代码建党 100 年百年风雨 ...
pytorch方法测试——卷积（二维）
pytorch方法测试--卷积(二维) 测试代码: import torch import torch.nn as nnm = nn.Conv2d(2, 2, 3, stride=2) input = ...
PyTorch安装测试训练建自己的数据集
Pytorch安装测试训练建自己的数据集前言一.PyTorch是什么? 二.PyTorch环境搭建 1.设备要求 2.安装Pytorch 3.验证PyTorch 二.CIFAR10测试 1.关于C ...
Pytorch ResNet源码学习
一,残差网络架构 1,残差学习单元上图左对应的是浅层网络(18层,34层),而右图对应的是深层网络(50,101,152). 1. 左图为基本的residual block,residual map ...
Pytorch resnet花朵识别（5种花）附完整代码
notebook运行结果图: 随机从各种花的图片集中抽取一定数量的图片因为设备限制,用所有的图片,图片太多如果用cpu跑太慢了用gpu太多图片会out of memory 所以用小部分图片,了解 ...
基于深度学习的车牌检测识别(Pytorch)(ResNet +Transformer）
车牌识别概述基于深度学习的车牌识别,其中,车辆检测网络直接使用YOLO侦测.而后,才是使用网络侦测车牌与识别车牌号. 车牌的侦测网络,采用的是resnet18,网络输出检测边框的仿射变换矩阵,可检 ...
pytorch方法测试——损失函数（CrossEntropyLoss）
import torch import torch.nn as nn import math loss = nn.CrossEntropyLoss() input = torch.randn(1, 5 ...
pytorch CrossEntropyLoss测试
数据类型只支持long类型 import torch import torch.nn as nn loss = nn.CrossEntropyLoss() # input, NxC=2x3 input ...
pytorch mseloss测试
0和1, 80和81的loss是一样的. 两个分布一样,loss为0 #双方都有0,会减少loss x值不用在 0-1之间求和: size_average=False mseloss测试 imp ...

PyTorch ResNet 测试

PyTorch ResNet 测试相关推荐

最新文章

热门文章