
Pytorch
chan.zc
DL & ML & SemanticSegmentation & Multimodal
展开
-
Pytorch中Softmax、Log_Softmax、NLLLoss以及CrossEntropyLoss的关系与区别详解
参考:https://blog.youkuaiyun.com/qq_28418387/article/details/95918829?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.noneca转载 2020-05-18 10:37:32 · 529 阅读 · 0 评论 -
Pytorch学习:多标签分类
多标签数据的dataloader的写法参考转载自: https://discuss.pytorch.org/t/multi-label-classification-in-pytorch/905/7转载 2020-03-17 16:33:11 · 4817 阅读 · 0 评论 -
pytorch学习:准备自己的图片数据
转载自:https://www.cnblogs.com/denny402/p/7512516.html转载 2020-03-17 16:30:40 · 182 阅读 · 0 评论 -
Pytorch对自定义模型加载预训练参数
在平常的各项图像比赛中通常不能直接使用Pytorch提供的预训练模型,需要更改模型结构.如果直接重新写一个模型而没有预训练参数的话,训练任务的效果就会很差.为了使用预训练的参数,而又使模型适合我们的需求,所以需要对已经加载参数的模型修改结构.下面我将介绍使用它的方法.Pytorch加载预训练模型对于使用torchvision中的网络,加载预训练参数的方法是from torch.utils i...转载 2019-12-13 16:33:00 · 1697 阅读 · 0 评论 -
Pytorch tensor维度理解
二维:import torchfrom torch.autograd import Variable##单位矩阵来模拟输入input = torch.ones(1, 5, 5)input = Variable(input)输出:tensor([[[1., 1., 1., 1., 1.], [1., 1., 1., 1., 1.], [1., 1....原创 2019-12-12 15:27:02 · 1078 阅读 · 0 评论