使用pytorch自带的网络结构

介绍

pytorch为我们提供了很多现成的网络结构,我们应该如何去使用呢?

我们可以看到我们使用resnet的时候会有许多现成的

当然vgg也一样,都有着许多现成的网络

使用模板

python 复制代码
# 使用pytorch中定义好的网络模型
import torch.nn as nn
# 导入自带的网络包
from torchvision import models

class resnet18(nn.Module):
    def __init__(self):
        super(resnet18, self).__init__()
        # pretrained加载预训练模型,无预训练模型会自动下载
        self.model = models.resnet18(pretrained=True)
        self.num_features = self.model.fc.in_features
        self.model.fc = nn.Linear(self.num_features, 10)

    def forward(self, x):
        out = self.model(x)
        return out

def pytorch_resnet18():
    return resnet18()

想要用其他的网络只需要把里面的

复制代码
resnet18改成自己想用的网络即可

然后就是

复制代码
nn.Linear(self.num_features, 10)里面的10改成自己分类任务里面的特征个数即可
相关推荐
摘星编程11 分钟前
智能体核心架构解析:感知-推理-行动的完整闭环
人工智能·智能体架构·感知系统·推理算法·行动控制
RAY_010415 分钟前
Python—数据容器
开发语言·python
June bug18 分钟前
【python基础】python和pycharm的下载与安装
开发语言·python·pycharm
二二孚日29 分钟前
自用华为ICT云赛道AI第一章知识点-机器学习的常见算法
人工智能·华为
聚客AI35 分钟前
🎯 RAG系统工业级部署指南:六步实现<3%幻觉率的问答系统
人工智能·langchain·llm
im_AMBER1 小时前
python实践思路(草拟计划+方法)
开发语言·python
站大爷IP1 小时前
Python与JSON:结构化数据的存储艺术
python
掘金一周1 小时前
Figma Dev Mode MCP:大人,时代变了 | 掘金一周7.10
前端·人工智能·mcp
大千AI助手1 小时前
陶哲轩:数学界的莫扎特与跨界探索者
人工智能·数学·机器学习·概率·人物·天才·陶哲轩