pytorch
文章平均质量分 69
I松风水月
个人主页:https://chasecjg.github.io/
展开
-
pytorch容器之nn.Sequential, nn.ModuleList, nn.ModuleDict介绍
pytorch容器之nn.Sequential, nn.ModuleList, nn.ModuleDict介绍原创 2023-06-08 16:46:10 · 1563 阅读 · 0 评论 -
__call__函数介绍
__call__方法介绍原创 2023-06-08 09:51:04 · 1150 阅读 · 0 评论 -
pytorch冻结网络模型参数(迁移学习)
pytorch冻结网络模型参数(迁移学习)原创 2023-04-12 20:05:11 · 1275 阅读 · 1 评论 -
apex混合精度训练教程
apex混合精度训练教程原创 2023-04-03 11:25:44 · 709 阅读 · 2 评论 -
张量tensor学习笔记
张量tensor学习笔记原创 2022-10-03 11:15:00 · 549 阅读 · 0 评论 -
BN层的一点总结
BN层的一点总结原创 2022-09-29 10:02:55 · 787 阅读 · 0 评论 -
准确率,精准率,召回率,真正率,假正率,ROC/AUC
准确率,精准率,召回率,真正率,假正率,ROC/AUC原创 2022-08-24 10:16:01 · 28559 阅读 · 1 评论 -
Pytorch/Tensor/Numpy中的广播机制——Broadcast
Pytorch/Tensor/Numpy中的广播机制——Broadcast原创 2022-08-14 14:30:20 · 2199 阅读 · 3 评论 -
pytorch中.data()与.item(),.numpy(),.detach(),.cuda(),.cpu()的作用
pytorch中.data()与.item(),.numpy(),.detach(),.cuda(),.cpu()的作用原创 2022-05-05 09:28:49 · 2412 阅读 · 0 评论 -
opencv,Image(PIL),skimage,matplotlib.image,matplotlib.pyplot读取图像的区别以及如何显示读取的图像
opencv,Image(PIL),skimage,matplotlib.image,matplotlib.pyplot读取图像的区别以及如何显示读取的图像原创 2022-05-02 16:47:32 · 3176 阅读 · 0 评论 -
pytorch结合albumentations在图像分割中的应用
pytorch和Albummentation在图像分类,分割,检测中的应用原创 2022-03-21 21:21:19 · 1151 阅读 · 0 评论 -
pytorch自定义数据集
pytorch自定义数据集原创 2022-03-06 10:44:56 · 2876 阅读 · 4 评论 -
pytorch模型加载与保存,并在保存的模型基础上进行训练
两种方法:1.只保存参数(官方推荐,消耗的存储空间较小),详细讲解此方法。2.保存整个模型结构一.只保存参数1.保存:方法一:torch.save(model.state_dict(), path)model:定义的模型的实例变量,如model = resnet(),path是模型保存的路径,如path = “./model.pth”,path = “./model.pkl”,path = “./model.tar”,一定要加上后缀。方法二:如果想保存某一次的训练参数和模型,那么可以使用原创 2022-03-01 10:19:31 · 976 阅读 · 0 评论