pytorch
文章平均质量分 77
花海!
这个作者很懒,什么都没留下…
展开
-
相关网址整理
该网址会不断更新,有错误或网址私聊补充python官网:https://www.python.org/packages网址:https://www.lfd.uci.edu/~gohlke/pythonlibs/原创 2020-06-30 22:39:19 · 333 阅读 · 0 评论 -
创建类似CIFAR10的数据集以及自制读取方式(根据官网的cifar10数据集文件创建dataset)
网上有很多用自己的数据集制作类似CIFAR10的数据集方法,但制作完了data_batch_1文件以后,想要直接训练自己的数据中间还差了将数据集文件制作成网络所需要的dataset的过程,我在网上找是没找到,如果有找到的还请告知一下,将自己的数据集制作成类似data_batch_1的文件参考https://blog.csdn.net/weixin_40437821/article/details/103603700从data_batch_1到dataset:...原创 2020-09-22 17:35:54 · 2017 阅读 · 4 评论 -
import apex出错,apex标红解决办法
运行一个pytorch时发现import apex标红,同时通过阿里云,清华源都无法下载,解决办法如下:作用:提升GPU上训练速度github下的网址为https://github.com/NVIDIA/apex,可以通过直接下载压缩包到自己的site-package文件夹下并解压,也可以通过命令行进行安装:Linux下安装:为了性能和完整的功能,建议通过CUDA和c++扩展来安装Apexclone之前现将地址cd到 /site-packages文件夹下,然后在site-packages文件夹下打原创 2020-08-17 23:46:10 · 5434 阅读 · 2 评论 -
关于nn.ReLU(inplace=True)和nn.ReLU(inplace=False)的区别
可以把inplace当作是一个进行relu运算后输入变化与否的开关,True代表输入改变,False输入保持原样。验证代码如下:>>> input = torch.randn(7)>>> inputtensor([ 0.1571, -0.6665, -0.5410, -0.6172, 0.1348, 0.4602, 0.0018])>>> m = nn.ReLU(inplace=True)>>> out_m = m(in原创 2020-08-16 20:16:17 · 5128 阅读 · 0 评论