坎坷复现之路

最新推荐文章于 2022-12-23 22:15:00 发布

掉毛学渣

最新推荐文章于 2022-12-23 22:15:00 发布

阅读量80

点赞数

分类专栏：学习笔记文章标签： pytorch

本文链接：https://blog.csdn.net/wh22141334/article/details/118441893

版权

学习笔记专栏收录该内容

9 篇文章 0 订阅

订阅专栏

第一句代码：
在这里插入图片描述
目的是在当前版本使用新版本功能，如2.7中使用3.6的功能

torch.utils.model_zoo.load_url(url, model_dir=None)
url：要下载对象的URL
model_dir：下载到本地文件地址
class MultiInputImages(models.ResNet):
属于类的继承，如果要编写的类是另一个现成类的特殊版本，可以采取继承父类的方法。
def __init__(self, block, layers, num_classes=1000, num_input_images=1): super(ResNetMultiImageInput, self).__init__(block, layers)
__init()__接受创建Resnet所需的信息，super()将父类与子类联结起来
resnet中常见的make_layer(),

    def _make_layer(self, block, planes, blocks, stride=1):
        downsample = None
        if stride != 1 or self.inplanes != planes * block.expansion:
            downsample = nn.Sequential(
                nn.Conv2d(self.inplanes, planes * block.expansion,
                          kernel_size=1, stride=stride, bias=False),
                nn.BatchNorm2d(planes * block.expansion),
            )

        layers = []
        layers.append(block(self.inplanes, planes, stride, downsample))
        self.inplanes = planes * block.expansion
        for i in range(1, blocks):
            layers.append(block(self.inplanes, planes))

        return nn.Sequential(*layers)