MTCNN以指定方式初始化权重

匿名 (未验证) 提交于 2019-12-02 23:43:01
 def weights_init(m):     if isinstance(m, nn.Conv2d) or isinstance(m, nn.Linear):#判断m对象是否属于这些类         nn.init.xavier_uniform(m.weight.data)         nn.init.constant(m.bias, 0.1)  class Net:     def __init():         self.apply(weights_init)#权重初始化

文章来源: https://blog.csdn.net/weixin_38241876/article/details/91958751
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!