![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch笔记本
文章平均质量分 70
Iron_lyk
中山大学智能工程学院
展开
-
模型权值共享/参数访问/初始化
模型权值共享/参数访问/初始化原创 2023-03-01 15:38:32 · 274 阅读 · 0 评论 -
【Pytorch踩坑记录】深度学习程序在训练第x个Epoch的过程中直接被退出终端
我在一个大的数据集上训练模型时,终端或vscode界面老是在程序在跑第3个epoch的过程中自动被卡退,注意不是程序报错,而是直接将vscode退出,连错误都不报这就是头疼之处,并不知道问题出现在哪里原创 2022-11-03 20:53:21 · 3278 阅读 · 3 评论 -
【Pytorch】自定义Dataset类按照index去除train_set或test_set中指定数据
【Pytorch】自定义Dataset类中按照index去除train_set或test_set中指定数据。已知原始数据或预处理数据中有些样本是有问题的,但是又不想简单粗暴的在传递给Dataset类之前做数据清理(删除掉),而是想要在自己写的Dataset类中自动过滤(去除)掉这些样本的数据原创 2022-09-24 16:54:05 · 1140 阅读 · 0 评论 -
【Pytorch】torch.nn.Dropout()的两种用法:防止过拟合 & 数据增强
Dropout方法是一种在训练模型时被广泛应用的trick,目的是防止模型过拟合,原理是使网络中某一层的每个参数以一定概率被mask(变为0),只用剩下的参数进行训练,从而达到防止模型过拟合的目的。...原创 2022-07-13 10:53:58 · 6945 阅读 · 0 评论 -
【Pytorch】autograd.Variable类详解 & tensor.backward()进行反向传播梯度计算过程
Variable类是autograd包中很重要的一类,它的作用是包装Tensor,将一个tensor其变成计算图中的一个节点(变量)。原创 2022-07-04 19:20:17 · 3723 阅读 · 1 评论