首页 > 其他 > 详细

pytorch学习-2

时间:2020-05-25 10:22:15      阅读:62      评论:0      收藏:0      [点我收藏+]

尝试修改网络结构的记录:

1、resnet18修改为vgg16,epoch=2时效果差点,修改:

model_conv = models.vgg16(pretrained=True)

#model_conv = models.restnet18(pretrained=True)

2、如果每个stage结构都一样,可以写如下,再传参数。

self.conv1 = nn.Sequential(
nn.Conv2d()
nn.BatchNormal()
nn.PReLU()
nn.MaxPooling()
nn.Dropout()
)

传参数:conv2d(n_in,n_out,kernel,stride,padding)

batchnormal(n_out)待处理数据的channel;batchnormal(n_out,0.1)包含了?

PRelu可以换成Relu,无参数;

Maxpooling(2)表示(2,2)的maxpooling

Dropout(0.2),网络小时取值也小,一般0.2--0.5

3、

技术分享图片

 

参考:

sequential的定义和参数传递:(定义函数)

https://www.debugger.wiki/article/html/157249224958066

sequential和调用

https://www.cnblogs.com/wangguchangqing/p/11058525.html

https://www.cnblogs.com/darkknightzh/p/6065526.html  sequential显示net的详细方法

https://blog.csdn.net/t20134297/article/details/104960101?utm_medium=distribute.pc_relevant.none-task-blog-baidujs-1

torch 的yolo4: https://blog.csdn.net/weixin_44791964/article/details/106214657

pytorch学习-2

原文:https://www.cnblogs.com/haiyanli/p/12954274.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!