PyTorch 中的 Tips
jinfeng2411
这个作者很懒,什么都没留下…
展开
-
全新 ubuntu18.04 + nvidia显卡配置 + cuda_10.0 + cudnn v7.4.1 + Anaconda_5.2.0 + pytorch 配置
最近新买了一块显卡,准备配个pytorch环境,网上教程一大堆,能用的、清晰明了的没几个,被迫重装了几次系统才搞定,在此记录一下,以防下次用到。全新ubuntu18准备工作:1、更换阿里源:https://blog.csdn.net/once_pluto/article/details/83384866(不然很多软件装得很慢或装不了)2、安装 build-essential(它能帮你装...原创 2019-11-15 09:19:08 · 232 阅读 · 0 评论 -
nn.AdaptiveAvgPool2d() 与 nn.AvgPool2d() 模块的区别
两者都是做二维的平均池化,但是它们关注的参数却不一样。1、nn.AvgPool2d() 模块模块原型为其中参数为一般我们使用它的时候,只需要关注 kernel_size 、stride 与 padding 三个参数就行了,最后输出的尺寸为:其中,x 表示输入的维度大小, y 表示对应输出的维度大小。2、nn.AdaptiveAvgPool2d() 模块...原创 2019-10-13 21:24:46 · 41089 阅读 · 6 评论