乐胖代购免代理版

安装torch后，torch.cuda.is_available()结果为false的问题

1. 在conda虚拟环境中安装了torch，一般命令都可以正常使用，但是使用cuda的命令to三个不同的版本，这里在虚

pytorch

python

深度学习

虚拟环境

官网

原创 2022-10-21 16:52:46 2557 阅读

Pytorch加载自己的数据集(使用DataLoader读取Dataset)

1. 我们经常可以看到Pytorch加载数据集会用到官方整理好的数据集。很多时候我们需要加载自己的数据集，这时候我们需要使用Dataset和DataLoaderDataset：是被封装进Dat

pytorch

数据

加载

封装

原创 2022-10-21 16:30:24 649 阅读

Expected object of scalar type Long but got scalar type Double for argument #2 ‘target‘

1.pytorch报错：loss_class = torch.nn.CrossEntropyLoss()s_data, s_label = data_source[0].to(DEVICE), data_sourc_data=s_data.float(), alpha=alph...

pytorch

深度学习

python

scala

数据类型

原创 2022-10-21 16:27:18 624 阅读

Pytorch的modle.train，model.eval，with torch.no_grad的个人理解

1. 最近在学习pytorch过程中遇到了几个问题，不理解为什么在训练和测试函数中model.eval()，和model.train()的区别，经查阅

数据

测试阶段

参考文档

原创 2022-10-21 16:26:33 667 阅读

pytorch使用ImageFolder和random_split读取和划分数据集

. 首先，如果数据集层级结构...

pytorch

深度学习

python

数据

数据集

原创 2022-10-21 16:14:36 565 阅读

pytorch中对于图像数据集进行划分成train，test和val

源文件

ide

目录结构

原创 2022-10-21 16:14:23 268 阅读

np.transpose()函数详解

1. 碰ose(1, 2, 0......

numpy

python

机器学习

数据

坐标轴

原创 2022-10-21 16:14:17 1039 阅读

pytorch定义新的自动求导函数

在pytorch中想自定义求导函数，通过实现torch.autograd.Function并重写forward和backward函数，来定义自己的自动求导运算。参考官网上的demo：传送门直接上代码，定义一个ReLu来实现自动求导import torchclass MyRelu(torch.autograd.Function): @staticmethod def forwar

随机数

反向传播

官网

原创 2022-10-21 16:04:22 166 阅读

官方博客	全部文章	热门标签	班级博客
了解我们	网站地图	意见反馈

鸿蒙开发者社区	51CTO学堂
51CTO	软考资讯

51CTO博客

mb63524b8b20f07的博客

安装torch后，torch.cuda.is_available()结果为false的问题

Pytorch加载自己的数据集(使用DataLoader读取Dataset)

Expected object of scalar type Long but got scalar type Double for argument #2 ‘target‘

Pytorch的modle.train，model.eval，with torch.no_grad的个人理解

pytorch使用ImageFolder和random_split读取和划分数据集

pytorch中对于图像数据集进行划分成train，test和val

np.transpose()函数详解

pytorch定义新的自动求导函数

欢迎