1060 416 batch_size 6
import os
import time
import torch
import torch.nn as nn
import torchvision.transforms as transforms
import torch.optim as optim
def conv3x3(in_planes, out_planes, stride=1, dilation=1):
"""3x3 convolution with padding"""
return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride, padding=dilation, bias=False,
dilation=dilation)
def conv1x1(in_planes, out_planes):
"""1x1 convolution"""
return nn.Conv2d(in_planes, out_planes, kernel_size=1, stride=1, bias=False)
class Linear(nn.Module):
def __init__(self):
super().__init__()
def forward(self, x):
return x
class BasicBlock(nn.Module):
expansion = 1
tran_expansion = 1
def __init__(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilatio