model.py 文件源码

python
阅读 26 收藏 0 点赞 0 评论 0

项目:planet-pytorch 作者: kefth 项目源码 文件源码
def __init__(self):
        super().__init__()
        self.pretrained_model = models.resnet18(pretrained=True)
        classifier = [
            nn.Linear(self.pretrained_model.fc.in_features, 17)
        ]
        self.classifier = nn.Sequential(*classifier)
        self.pretrained_model.fc = self.classifier
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号