- 博客(8)
- 收藏
- 关注
原创 从Segmentation到Classification任务:数据整理
我本来想尝试图像分割(image segmentation task),老大觉得还我还需要打基础,建议从更简单的图像分类(classification task)先做起。我就,当然听老大的。
2024-02-04 10:36:49
487
1
原创 PyTorch中class的使用
我们来了解一下PyTorch中类的基本结构及其典型用例。# 在构造函数中定义层或组件# 在这个方法中定义前向传播或计算return x# 实例化模型# 示例输入张量# 前向传播接下来,逐行解释代码帮助记忆。在这行代码中,我们定义了名为 SimpleModel 的新类,它继承自 nn.Module。通过继承自 nn.Module,SimpleModel 获得了在深度学习环境中非常有用的一些功能和特性。这也是创建神经网络模型的常见做法。
2024-02-04 07:08:02
2216
2
原创 pytorch中如何存储模型结构及权重文件
将模型结构及参数都保存起来,在后续模型加载时不需要预先定义模型的网络结构,这种方法使用方便,但耗内存。具体方法如下:1) 保存。
2024-02-01 09:16:54
975
1
原创 ubuntu 系统下 VSCode(python)无法启动代码补全
新手,有深度学习的基础,但刚接触pytorch(是的,我也不知道为什么自己一开始选了tensorflow)。需要代码补全。
2024-01-08 17:52:41
560
原创 双系统下启动Ubuntu 无征兆出现启动后黑屏只有左上角有一个下划线不停闪烁
Ubuntu 新手上路,熟悉总要有个过程,启动Ubuntu后黑屏同时左上角有个下划线不停闪烁。这个问题之前遇到过一次,今天又出现了,好记性不如烂笔头,所以写下来。问题: 我不清楚为什么会出现这个问题,但是屡试不爽。
2024-01-05 09:53:08
3169
5
空空如也
混淆矩阵TN计算的问题
2022-08-29
TA创建的收藏夹 TA关注的收藏夹
TA关注的人