![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
MaloFleur
这个作者很懒,什么都没留下…
展开
-
pytorch使用出现"RuntimeError: An attempt has been made to start a new process before the..." 解决方法
今天在运行Pytorch手写数字识别代码时,出现了如下错误:其实挺奇怪的… 因为这段代码是前几天刚写完跑成功的,今天再跑就出问题了至于解决方法也很简单,出错提到了是多线程的缘故,那么就有如下两种:去掉num_workers参数# train_loader = Data.DataLoader(dataset=train_data, batch_size=BATCH_SIZE, shuf...原创 2018-09-18 10:18:41 · 31458 阅读 · 21 评论 -
pytorch中与维度/变换相关的几个函数
torch.size ()先说torch.size()函数,因为后面的方法都会用这个方法看到变换后的矩阵的维度通过该方法,可以查看当前Tensor的维度,用法也很简单:torch.view()torch.permute()torch.squeeze() / torch.unsqueeze()...原创 2019-01-11 22:59:15 · 32606 阅读 · 0 评论 -
pytorch 使用visdom进行可视化
相比tensorbordX, visdom刷新更快,界面体验也良好,首先是visdom的安装,与普通的python库一样,直接pip install visdom即可成功安装后,在控制台下输入python -m visdom.server复制http://localhost:8097,输入浏览器即可打开visdom可视化窗口调用visdom主要需要如下语句:from visdom imp...原创 2019-05-07 19:54:14 · 936 阅读 · 0 评论 -
pytorch源码解读——RNN/LSTM篇
文章的字母中:b: batch_sizet: time_stepn: num_featureh: hidden_size假设输入数据维度input = (b, t, n)所设计的LSTM模型如下:class MYLSTM(nn.Module): def __init__(self, input_size, hidden_size, out_size): s...原创 2019-06-05 22:10:41 · 9266 阅读 · 7 评论 -
Pytorch多机多卡训练
本文转载自知乎:Pytorch多机多卡训练 - 谜一样的男子的文章 - 知乎关于Pytorch分布训练的话,大家一开始接触的往往是DataParallel,这个wrapper能够很方便的使用多张卡,而且将进程控制在一个。唯一的问题就在于,DataParallel只能满足一台机器上gpu的通信,而一台机器一般只能装8张卡,对于一些大任务,8张卡就很吃力了,这个时候我们就需要面对多机多卡分布式训练这...转载 2019-06-20 21:17:53 · 14678 阅读 · 4 评论