花费 30 ms
PyTorch环境配置及安装

环境配置 温馨提示:为了更好的教程体验,提供视频、阅读地址 Youtube: https://www.youtube.com/playlist?list=PLgAyVnrNJ96CqYdjZ8 ...

2019-11-03 05:01 4 59023
Pytorch中的torch.cat()函数

cat是concatnate的意思:拼接,联系在一起。 先说cat( )的普通用法 如果我们有两个tensor是A和B,想把他们拼接在一起,需要如下操作: ...

2018-12-23 04:29 0 43400
Pytorch里的CrossEntropyLoss详解

在使用Pytorch时经常碰见这些函数cross_entropy,CrossEntropyLoss, log_softmax, softmax。看得我头大,所以整理本文以备日后查阅。 首先 ...

2019-02-19 23:12 0 38716
torch.utils.data.DataLoader使用方法

数据加载器,结合了数据集和取样器,并且可以提供多个线程处理数据集。在训练模型时使用到此函数,用来把训练数据分成多个小组,此函数每次抛出一组数据。直至把所有的数据都抛出。就是做一个数据的初始化。 ...

2019-03-30 03:07 1 27589
pytorch: 准备、训练和测试自己的图片数据

大部分的pytorch入门教程,都是使用torchvision里面的数据进行训练和测试。如果我们是自己的图片数据,又该怎么做呢? 一、我的数据 我在学习的时候,使用的是fashion-mnist。 ...

2017-09-14 20:54 8 55239
Pytorch中nn.Conv2d的用法

Pytorch中nn.Conv2d的用法 nn.Conv2d是二维卷积方法,相对应的还有一维卷积方法nn.Conv1d,常用于文本数据的处理,而nn.Conv2d一般用于二维图像。 先看一下接口定 ...

2019-11-28 01:36 0 12199
源码详解Pytorch的state_dict和load_state_dict

在 Pytorch 中一种模型保存和加载的方式如下: model.state_dict()其实返回的是一个OrderDict,存储了网络结构的名字和对应的参数,下面看看源代码如何实现的。 st ...

2019-12-21 06:31 2 26204
pytorch中的cat、stack、tranpose、permute、unsqeeze

Cat 对数据沿着某一维度进行拼接。cat后数据的总维数不变. 比如下面代码对两个2维tensor(分别为2*3,1*3)进行拼接,拼接完后变为3*3还是2维的tensor。 import tor ...

2018-08-01 08:25 1 24925
一文弄懂Pytorch的DataLoader, DataSet, Sampler之间的关系

以下内容都是针对Pytorch 1.0-1.1介绍。 很多文章都是从Dataset等对象自下往上进行介绍,但是对于初学者而言,其实这并不好理解,因为有的时候会不自觉地陷入到一些细枝末节中去,而 ...

2019-08-06 22:51 5 12016

 
粤ICP备18138465号  © 2018-2021 CODEPRJ.COM