pytorch
霜溪
专注于算法和编程
展开
-
卷积神经网络|猫狗分类系列--导入kaggle猫狗数据集
解决任何真实问题的重要一步是获取数据,Kaggle提供了大量不同数据科学问题的竞赛。原创 2024-01-08 20:08:53 · 1189 阅读 · 0 评论 -
PyTorch|一些简单操作
在使用PyTorch构建自己的神经网络时,灵活度非常大,这给了用户更多的发挥空间,这里介绍一些简单的操作。原创 2024-01-06 17:12:48 · 865 阅读 · 0 评论 -
PyTorch|PyTorch张量解释
神经网络中的输入、输出和转换都使用张量表示,因此,神经网络编程大量使用张量,张量是我们在 PyTorch 中编程神经网络时将使用的数据结构。原创 2024-01-04 17:16:16 · 545 阅读 · 0 评论 -
PyTorch|在张量运算中使用GPU
例如,假设我们写的代码到处使用cuda()方法,然后,我们把代码交给一个没有GPU的用户,这样做是行不通的。pytorch在张量运算时允许我们在GPU上进行计算,我们可以采用这些方法将数据无缝的移入或移出GPU。总结起来,在我们编写的代码中,尽量使用这种形式:。对此,我们可以这样处理:。,当我们进入CPU时,我们可以使用。当我们进入GPU时,我们可以使用。方法,去GPU的时候,我们写到。,去CPU的时候,我们写到。同时,我们还可以使用。数据必须移到GPU上。原创 2024-01-04 16:46:12 · 767 阅读 · 0 评论