Pytorch
文章平均质量分 58
Pytorch Examples
挪威的深林
深度学习 (Deep learning),机器学习(Machine learning),智能控制(Intelligent control),机器人(Robotics),RGB-D目标感知(RGB-D object perception)
展开
-
【Pytorch 第四讲】图像分类的Tricks
在分类问题中,最后一层一般是全连接层,然后对应标签的one-hot编码,即把对应类别的值编码为1,其他为0。这种编码方式和通过降低交叉熵损失来调整参数的方式结合起来,会有一些问题。这种方式会鼓励模型对不同类别的输出分数差异非常大,或者说,模型过分相信它的判断。但是,对于一个由多人标注的数据集,不同人标注的准则可能不同,每个人的标注也可能会有一些错误。模型对标签的过分相信会导致过拟合。原创 2024-03-01 20:27:04 · 1106 阅读 · 0 评论 -
【Pytorch 第三讲】如何使用pre-trained weights 来训练自己的模型
理由#有时在训练自己的模型时,如果从头开始训练自己的模型, 不但费时费力, 有时可能训练了很久, 好不容易收敛, 发现结果不是太好。如果能够基于被人 已经训练好的权重, 初始化自己的模型。那么在训练自己的模型时会事半功倍。原创 2024-01-26 20:47:38 · 934 阅读 · 0 评论 -
【Pytorch 第二讲】 如何遍历 或者查看Model权重/参数/tesnsor_size/dict_names
【代码】【Pytorch 第二讲】 如何遍历 或者查看Model权重/参数/tesnsor_size/dict_names。原创 2024-01-25 21:16:14 · 586 阅读 · 0 评论 -
【Pytorch 第一讲】 如何加载预训练模型
导入 PyTorch# 初始化你的模型# 加载预训练权重# 或者当完成这个加载以后, 可以考虑打开 checkpoint , 看看该模型保存时,包含哪些dict keys.结果: dict_keys(['epoch', 'arch', 'state_dict',根据上面的 state_dict, 可以接下来用来将权重赋予模型Model# 将权重赋值给模型。原创 2024-01-25 21:10:52 · 1330 阅读 · 0 评论