pytorch
文章平均质量分 84
来自新世界、
这个作者很懒,什么都没留下…
展开
-
U-net(2)
Unet网络:利用DoubleConv, Down, Up, OutConv四个模块组装U-net模型,其中Up即右侧模型块之间的上采样连接(Up sampling)部分,注意U-net的跳跃连接(Skip-connection)也在这部分(torch.cat([x2, x1], dim=1))。因为每个子块内部的两次卷积(Double Convolution),所以上采样后也有DoubleConv层。PyTorch模型定义的方式1.Module 类是 torch.nn 模块里提供的一个模型构造类 (n原创 2022-03-30 19:51:40 · 2962 阅读 · 2 评论 -
pytorch基础(3)【sigmoid,激活函数,优化器】
sigmoid函数logistic function是一种sigmoid函数和Linear的区别sigmoid函数是不需要参数的,所以不用对其初始化(直接调用nn.functional.sigmoid即可)。另外loss函数从MSE改用交叉熵BCE:尽可能和真实分类贴近。# -*- coding: utf-8 -*-"""Created on Mon Oct 18 08:35:00 2021@author: 86493"""import torchimport torch.nn原创 2022-03-27 22:20:14 · 3600 阅读 · 0 评论 -
pytorch基础(2)【代码例子】
线性模型# -*- coding: utf-8 -*-import numpy as npimport matplotlib.pyplot as pltx_data = [1.0, 2.0, 3.0]y_data = [2.0, 4.0, 6.0]def forward(x): return x * wdef loss(x, y): y_pred = forward(x) return (y_pred - y) * (y_pred - y)# 保存权重w_l原创 2022-03-27 20:43:53 · 1325 阅读 · 0 评论