pytorch
文章平均质量分 71
googler_offer
这个作者很懒,什么都没留下…
展开
-
pytorch-Mnist教程
pytorch-Mnist教程https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/#blas-and-lapack-operationshttps://github.com/pytorch/examples/blob/master/mnist/main.py资源的两个链接:第一个是中文文档,第二个是g...原创 2018-10-19 01:22:59 · 2327 阅读 · 0 评论 -
pytorch-nn.xx与nn.functional.xx的区别
两者的相同之处:nn.Xxx和nn.functional.xxx的实际功能是相同的,即nn.Conv2d和nn.functional.conv2d 都是进行卷积,nn.Dropout 和nn.functional.dropout都是进行dropout,。。。。。; 运行效率也是近乎相同。nn.functional.xxx是函数接口,而nn.Xxx是nn.functional.xxx的类封装...原创 2019-04-01 09:23:02 · 1623 阅读 · 0 评论 -
pytorch基础教程-模型编写(一)
'''pytorch 中神经网络模块化接口的了解1, torch.nn是专门为设计的模块化接口,nn构建位于autograd之上,可以定义和运行神经网络2, nn.Module是nn中十分重要的类,包含网络各层的定义以及forward方法''''''定义自己的网络:1, 需要继承nn.Module类,并实现forward方法2, 一般把网络中具有可学习参数的层放在构...原创 2019-04-01 10:29:12 · 982 阅读 · 0 评论 -
pytorch基础教程-模型编写(二)
对于一些重复的layer可以采用nn.Sequential先进行封装layer,然后调用_make_layer_进行实现以ResNet18/34为例模型图:ResNet中将一个跨层直连的单元叫做Residual block,其结构如下,当输入和输出的通道不一致的时候,需要一个专门的单元将二者转成一致,使其可以相加。实现:class BasicBlock(nn.Modu...原创 2019-04-01 10:53:39 · 1965 阅读 · 0 评论