目录
PyTorch ResNet 测试
原文:
PyTorch实现的ResNet50、ResNet101和ResNet152_mingo_敏的博客-CSDN博客_resnet50和resnet101区别
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 time
import torch.nn as nn
from torch.nn.parameter import Parameter
import torch
def conv3x3(in_plan