上一页 1 ··· 12 13 14 15 16 17 18 19 20 ··· 36 下一页

2020年6月15日

利用Pytorch DataLoader中的collate_fn自定义批加载方式

摘要: Pytorch DataLoader中的collate_fn允许用户灵活的定义批样本的加载方式,例如加载变尺寸的张量。 collate_fn (callable, optional) – merges a list of samples to form a mini-batch of Tensor( 阅读全文

posted @ 2020-06-15 22:31 那抹阳光1994 阅读(1253) 评论(0) 推荐(0) 编辑

2020年6月10日

sklearn之分类模型混淆矩阵和分类报告(转载)

摘要: 机器学习中的分类问题评估模型性能时,往往需要计算各种评价指标。通过计算混淆矩阵(confusion matrix)可以方便地导出各种指标,例如precision(查准率)、recall(tpr)(查全率、召回)、accuracy、fpr、F1分数、Roc曲线、Auc等。 一些常用的分类评价指标计算公 阅读全文

posted @ 2020-06-10 18:34 那抹阳光1994 阅读(3377) 评论(0) 推荐(0) 编辑

2020年6月7日

Pytorch MSELoss

摘要: CLASS torch.nn.MSELoss(size_average=None, reduce=None, reduction='mean') Creates a criterion that measures the mean squared error (squared L2 norm) be 阅读全文

posted @ 2020-06-07 13:50 那抹阳光1994 阅读(1337) 评论(0) 推荐(0) 编辑

2020年6月5日

pytorch 中的view、reshape、permute、transpose、contiguous

摘要: pytorch 中的view、reshape、permute、transpose、contiguous 1、 contiguous https://stackoverflow.com/questions/48915810/pytorch-contiguous https://blog.csdn.ne 阅读全文

posted @ 2020-06-05 20:34 那抹阳光1994 阅读(1060) 评论(0) 推荐(0) 编辑

pytorch教程之nn.Sequential类详解——使用Sequential类来自定义顺序连接模型

摘要: 前言:类似于keras中的序贯模型,当一个模型较简单的时候,我们可以使用torch.nn.Sequential类来实现简单的顺序连接模型。这个模型也是继承自Module类的,关于这个类,后面的文章会详细介绍。 一、关于Sequential类的简介先来看一下它的定义吧,在之前,我们首先需要明确一个特别 阅读全文

posted @ 2020-06-05 16:38 那抹阳光1994 阅读(4127) 评论(0) 推荐(2) 编辑

Pytorch checkpoint

摘要: checkpoint一种用时间换空间的策略 torch.utils.checkpoint.checkpoint(function, *args, **kwargs) 为模型或模型的一部分设置Checkpoint 。 检查点用计算换内存(节省内存)。 检查点部分并不保存中间激活值,而是在反向传播时重新 阅读全文

posted @ 2020-06-05 14:48 那抹阳光1994 阅读(5761) 评论(0) 推荐(1) 编辑

2020年6月3日

Pytorch出现 raise NotImplementedError

摘要: 问题所在:def forward多了一个Tap位,它与def __init__并不对齐。 解决方案:删除def forward多出的Tap位,将其与def __init__对齐即可。 果然,我还是一个不落的犯了所有奇葩错误。。。[Dog] 阅读全文

posted @ 2020-06-03 20:50 那抹阳光1994 阅读(1271) 评论(0) 推荐(1) 编辑

pytorch nn.Module类—使用Module类来自定义模型

摘要: 前言 pytorch中对于一般的序列模型,直接使用torch.nn.Sequential类及可以实现,这点类似于keras,但是更多的时候面对复杂的模型,比如:多输入多输出、多分支模型、跨层连接模型、带有自定义层的模型等,就需要自己来定义一个模型了。本文将详细说明如何让使用Mudule类来自定义一个 阅读全文

posted @ 2020-06-03 20:47 那抹阳光1994 阅读(3511) 评论(0) 推荐(0) 编辑

pytorch学习

摘要: 简洁易懂的博客总结 来自博客专栏:https://blog.csdn.net/dss_dssssd/column/info/29714 pytorch系列 1 python class 中 的__call__方法 https://blog.csdn.net/dss_dssssd/article/de 阅读全文

posted @ 2020-06-03 14:37 那抹阳光1994 阅读(259) 评论(0) 推荐(0) 编辑

Pytorch中的nn.Sequential

摘要: A sequential container. Modules will be added to it in the order they are passed in the constructor. Alternatively, an ordered dict of modules can als 阅读全文

posted @ 2020-06-03 14:13 那抹阳光1994 阅读(2171) 评论(0) 推荐(0) 编辑

上一页 1 ··· 12 13 14 15 16 17 18 19 20 ··· 36 下一页

导航