PyTorch
文章平均质量分 66
PyTorch的学习
条件漫步
这个作者很懒,什么都没留下…
展开
-
PyTorch RuntimeError: DataLoader worker (pid(s) 15804) exited unexpectedly
@创建于:2021.10.201、报错RuntimeError: DataLoader worker (pid(s) 15804) exited unexpectedly网络上给出了方案,如:cuda 虚拟环境的共享内存不足,增大内存;改成更小的batchsize。2、我的方案我的代码中,num_workers = 2,启用了多进程。Python中多进程(multiprocessing这个模块包)的内容必须放在if name == 'main’之内才可以。多线程(threading这个模原创 2021-10-20 17:17:52 · 3186 阅读 · 0 评论 -
PyTorch FashionMNIST时装分类 代码实践
@创建于:2021.10.20使用FashionMNIST数据集(https://github.com/zalandoresearch/fashion-mnist ),对10个类别的“时装”图像进行分类。FashionMNIST数据集中包含已经预先划分好的训练集和测试集,其中训练集共60,000张图像,测试集共10,000张图像。每张图像均为单通道黑白图像,大小为32*32pixel,分属10个类别。import osimport numpy as npimport pandas as pdim原创 2021-10-20 17:09:52 · 1686 阅读 · 0 评论 -
PyTorch里的损失函数(loss) 和 优化器(optimizer)
@创建于:2021.10.19文章目录1、损失函数1.1 二分类交叉熵损失函数1.2 交叉熵损失函数1.3 L1损失函数1.4 平滑L1 (Smooth L1)损失函数1.5 MSE损失函数1.6 目标泊松分布的负对数似然损失1.7 KL散度1.8 MarginRankingLoss1.9 多标签边界损失函数1.10 二分类损失函数1.11 多分类的折页损失1.12 三元组损失1.13 HingEmbeddingLoss1.14 余弦相似度1.15 CTC损失函数2、优化器3、参考链接1、损失函数本转载 2021-10-19 15:39:27 · 1771 阅读 · 0 评论 -
并行计算简介
@创建于:2021.10.15文章目录1 为什么要做并行计算2 CUDA是个啥3 做并行的方法3.1 **网络结构分布到不同的设备中(Network partitioning)**3.2 **同一层的任务分布到不同数据中**(**Layer-wise partitioning**)3.3 **不同的数据分布到不同的设备中,执行相同的任务(Data parallelism)**4 参考链接在利用PyTorch做深度学习的过程中,可能会遇到数据量较大无法在单块GPU上完成,或者需要提升计算速度的场景转载 2021-10-15 17:16:37 · 698 阅读 · 0 评论 -
PyTorch的张量知识与梯度计算,仅代码
@创建于:2021.10.15原文介绍参见 thorough-pytorch/第二章 PyTorch基础知识/# -*- coding:UTF-8 -*- # datetime: 2021/10/12 16:48# software: PyCharm""" 文件说明: 学习张量知识 https://github.com/datawhalechina/thorough-pytorch """from __future__ import print_functionimport tor原创 2021-10-15 17:11:50 · 131 阅读 · 0 评论 -
PyTorch简介与资源
PyTorch的介绍PyTorch是由Facebook人工智能研究小组开发的一种基于Lua编写的Torch库的Python实现的深度学习库,目前被广泛应用于学术界和工业界,而随着Caffe2项目并入Pytorch, Pytorch开始影响到TensorFlow在深度学习应用框架领域的地位。总转载 2021-10-12 16:09:47 · 1579 阅读 · 0 评论