![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch学习与使用
文章平均质量分 59
daydayjump
计算机小白,正奔跑在学习的路上。
展开
-
pytorch使用记录(一) Transfer Learning tutorial | 如何使用预训练模型VGG16_bn
官方文档:Transfer Learning tutorial迁移学习(Transfer Learning) 迁移学习是深度学习中很重要的一点。本身是指将已经训练好的模型参数迁移到新的模型,s...原创 2018-06-27 13:44:53 · 5063 阅读 · 0 评论 -
pytorch使用记录(二) 参数初始化
本文主要记录如何在pytorch中对卷积层和批归一层权重进行初始化,也就是weight和bias。主要会用到torch的apply()函数。【apply】apply(fn):将fn函数递归地应用到网络模型的每个子模型中,主要用在参数的初始化。使用apply()时,需要先定义一个参数初始化的函数。def weight_init(m): classname = m.__class__.__na...原创 2018-07-03 15:37:30 · 12868 阅读 · 3 评论 -
pytorch使用记录(三) 多GPU训练
在具体使用pytorch框架进行训练的时候,发现实验室的服务器是多GPU服务器,因此需要在训练过程中,将网络参数都放入多GPU中进行训练。 正文开始: 涉及的代码为torch.nn.DataParallel,而且官方推荐使用nn.DataParallel而不是使用multiprocessing。官方代码文档如下:nn.DataParallel 教程文档如下:tutorial...原创 2018-07-22 23:06:17 · 55300 阅读 · 23 评论 -
pytorch使用记录(四)1.0版本安装时问题记录
pytorch的更新速度还是很快的,现在已经出现了1.0版本,提供了C++接口。而且还有许多性能的更新,大有赶超TF的趋势。现在十分庆幸当初选择使用pytorch神经网络框架。 本文主要是记录安装过程中的一个问题。因为之前已经安装了pytorch 0.4.1,所以这次算是升级。直接在anaconda上输入命令conda install pytorch torchvision...原创 2018-12-11 21:47:03 · 5013 阅读 · 4 评论 -
pytorch使用记录(五) 关于tensor、PIL以及numpy转换的问题
最近在运行程序的时候一直出现如下错误: File "/home/daydayjump/Glow/glow/trainer.py", line 172, in train self.writer.add_image("1_prob/{}".format(bi), plot_prob([y_pred[bi], y_true[bi]], ["pred", "true"]).cuda()....原创 2019-03-26 00:30:13 · 6803 阅读 · 3 评论 -
pytorch使用记录(六) 明确使用哪块GPU
可以用在多GPU的服务器上,明确使用的是哪块GPU,可以将信息打印出来,返回的是device的列表。具体用到的pytorch函数是torch.cuda.device_count(),返回可得到的GPU数量。接下来是具体函数代码,分为两部分,一个主函数get_proper_device明确使用GPU还是CPU,另外一个函数get_proper_cuda_device是明确具体使用哪块GPU。...原创 2019-05-09 10:47:13 · 3501 阅读 · 0 评论