资料下载
pytorch如何构建网络模型
香香技术员
分享资料个
利用 pytorch 来构建网络模型有很多种方法，以下简单列出其中的四种。
假设构建一个网络模型如下：
卷积层--》Relu 层--》池化层--》全连接层--》Relu 层--》全连接层
首先导入几种方法用到的包：
import torch
import torch.nn.functional as F
from collections import OrderedDict
第一种方法
# Method 1 -----------------------------------------
class Net1（torch.nn.Module）：
def __init__（self）：
super（Net1， self）.__init__（）
self.conv1 = torch.nn.Conv2d（3， 32， 3， 1， 1）
self.dense1 = torch.nn.Linear（32 * 3 * 3， 128）
self.dense2 = torch.nn.Linear（128， 10）
def forward（self， x）：
x = F.max_pool2d（F.relu（self.conv（x））， 2）
x = x.view（x.size（0）， -1）
x = F.relu（self.dense1（x））
声明：本文内容及配图由入驻作者撰写或者入驻合作网站授权转载。文章观点仅代表作者本人，不代表电子发烧友网立场。文章及其配图仅供工程师学习之用，如有内容侵权或者其他违规问题，请联系本站处理。 举报投诉
- 相关下载
- 相关文章
