pytorch
文章平均质量分 52
chentao326
这个作者很懒,什么都没留下…
展开
-
【神经网络 | 深度学习】pytorch快速搭建CNN网络
定义网络类 每一层卷积层由卷积核、池化和激活函数组成 一般情况下最后一层是全连接层 def __init__(self): super(CNNNET,self).__init__() self.con1 = nn.Sequential( nn.Conv2d(3,64,3,1,1), #3*128*128-->64*128*128 #3个输入,64输出(即卷积核),卷积核大小为3*3或者(3,3),原创 2021-05-31 18:01:48 · 328 阅读 · 0 评论 -
【神经网络 | 深度学习】pytorch快速搭建网络
快速搭建 导入包 torch.nn包含了大量和网络相关的方法 import torch import torch.nn as nn 使用GPU计算 device = torch.device('cuda' if torch.cuda.is_available else 'cpu') device = torch.device('cuda:0') 用第一块显卡计算,就一块显卡的话,传入'cuda' 设立输入和输出 torch.Tensor()直接建立张量, .to(device)表示使用CPU或者GPU计算原创 2021-05-31 17:50:33 · 186 阅读 · 1 评论