![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
各种码
aleien1
终身学习着的终身学习者
展开
-
pytorch的网络等转移到cuda
神经网络一般用GPU来跑,我们的神经网络框架一般也都安装的GPU版本,本文就简单记录一下GPU使用的编写。GPU的设置不在model,而是在Train的初始化上。第一步是查看是否可以使用GPUself.GPU_IN_USE = torch.cuda.is_available()就是返回这个可不可以用GPU的函数,当你的pytorch是cpu版本的时候,他就会返回False。然后是:s...原创 2019-11-13 15:53:03 · 2764 阅读 · 1 评论 -
pytorch loss function 总结
https://blog.csdn.net/zhangxb35/article/details/72464152这篇作者总结了pytorch手册中的损失函数,方便查找!转载 2019-11-13 14:28:19 · 97 阅读 · 0 评论 -
pytorch参数初始化细节以及验证是否成功
当把网络结构,如残差快,espcn块写好以后,就要开始进行参数初始化了。pytorch的参数初始化weight_init一般放在NET类里定义。class Net(nn.Module): def __init__(self, num_channels, base_channels, num_residuals, upscale_factor): super(Net, s...原创 2019-11-11 13:22:29 · 481 阅读 · 0 评论 -
关于pytorch写网络基础
首先所有网络都要继承nn.Module,不论是残差块还是网络主体。然后写网络的时候涉及到几个简单的函数,没啥说的,简单提几句。第一个函数时卷积函数,这里只说2d的卷积nn.Conv2d(in_channel,out_channel, kernel,stride,padding)#其中in_channel和out_channel分别代表输入、输出的网络通道数另一个函数是bn层nn.Ba...原创 2019-11-08 12:46:33 · 108 阅读 · 0 评论 -
python 迭代器,魔方方法,isinstance
今天介绍点牛逼的(相对昨天而言)。上午再看pytorch的Dataset、DataLoader、DataLoaderIter的相关博客,就是这篇很简单明了,然后我发现文章在介绍DataLoader的时候这样说:DataLoader返回的是一个迭代器,我们通过这个迭代器来获取数据我就奇怪,这个迭代器是个啥(毕竟本人学python时只看到了类,果然学习要坚持和深究)?找了下这篇文章,简单来...原创 2019-11-07 14:23:19 · 175 阅读 · 0 评论 -
pytorch写网络时的检查方法以及tensor转图片
先说第一个,刚刚接触pytorch,自己写网络感觉没人教感觉不容易,所以哪怕写完一个小的函数也要自己设置输入,去验证一下。前面读取数据的小函数写的确实没啥毛病,但是当我在pytorch里要用到Dataset的时候一写就牵扯到很多函数,好像没办法验证我这个写的有什么问题,或者说不明白我一个理想输入进去,输出是个啥,如果输出不对应那不是白忙活了。然后我在引用我重写的Dataset类的时候发现pyc...原创 2019-11-06 17:00:38 · 382 阅读 · 0 评论 -
pycharm 从一个脚本import同等级目录另一个脚本失败原因
昨天遇到的,觉得还是写一下,以后自己方便找,能帮助到各位当然更开心。就是在同等级目录中,pycharm从一个文件import另一个文件的函数会失败。尝试了很多方法,网上说了两种:重装numpy,还下载了一个什么dll文件在pycharm中,右击这两个文件的上一级目录,然后找到Mark Dirctory as,然后选sources root嗯,然后我都试过了,每个鸟用,很生气。我的目录...原创 2019-11-06 15:25:57 · 388 阅读 · 0 评论 -
pytorch 利用torchversion处理图像各种编译不通过
出现的编译报错有以下几种:TypeError: img should be PIL Image. Got <class 'numpy.ndarray'>'int' object is not iterable torchversion类似的报错,找了很多博客都改不了。我的程序主体是这样子:def input_data_process(crop, scale): r...原创 2019-11-06 10:36:35 · 628 阅读 · 0 评论 -
python历遍文件夹下所有文件
需求:想把一个文件夹下的所有文件名写到一个list里面一开始想用enumerate这个函数来搞定的,然后发现这个破函数的功能是将一个list、srt或者tup分割成索引序列。所以查资料发现有很多种方法,我试了一下os.walk,发现这个函数要写双循环,试了一下让我的函数进入了死循环,不知道什么情况。然后果断换方法。os.listdir(filenames_dir),直接返回一个文件名的lis...原创 2019-11-05 16:18:25 · 574 阅读 · 0 评论 -
argparse用法介绍
这个库用于在脚本中方便的改变参数值,简单地说有几步必要做的:import argparse声明一个命令行解析器:名字很高大上其实就一句话添加参数,同时设置参数格式利用parse_args获取所有参数这样就结束了,来看代码:import argparseparser = argparse.ArgumentParser(description="i am fine, really f...原创 2019-11-05 14:01:00 · 327 阅读 · 0 评论