72fc481
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import torch.nn as nn import torch.nn.functional as F from base import BaseModel from .ResNet_Zoo import ResNet, BasicBlock def resnet34(num_classes=10): return ResNet(BasicBlock, [3,4,6,3], num_classes=num_classes)
