IR-Net 测试

 

IR-Net

https://github.com/htqin/IR-Net

640 gpu测试50多ms

import time

import torch
import torch.nn as nn
import math
import torch.utils.model_zoo as model_zoo
import sys
sys.path.append('./modules')
import ir_1w32a

BN = None

__all__ = ['ResNet', 'resnet18', 'resnet34']


model_urls = {
    'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth',
    'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth',
}


def conv3x3Binary(in_planes, out_planes, stride=1):
    "3x3 convolution with padding"
    return ir_1w32a.IRConv2d(in_planes, out_planes, kernel_size=3, stride=stride,
                    padding=1, bias=False)


class BasicBlock(nn.Module):
    expansion = 1

    def __init__(self,