depth 44
1*3*512*512 1070 12ms
tx2上56跑不起来,44的 跑不动
import os
import math
import time
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable
cfg = [20, 32, 44, 56, 110, 164, 1001]
def conv3x3(in_channels, out_channels, stride=1):
return nn.Conv2d(in_channels, out_channels, kernel_size=3, stride=stride,
padding=1, bias=False)
class PreActBlock(nn.Module):
expansion &