Pytorch 通过hook获取网络中间结果(feature)
PART 1 定义网络结构 ResNet-18def conv3x3(in_planes, out_planes, stride=1): "3x3 convolution with padding" return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride, padding=1, bias=False)class BasicBlock(nn.Modul..
原创
2020-08-11 13:36:14 ·
2560 阅读 ·
3 评论