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