![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytoch实现
qq_43152949
这个作者很懒,什么都没留下…
展开
-
ranger 以及Optimizer相关实现
以ssd的train.py为例这一行将模型net的parameters导入优化器中, 还定义了其他参数类Optimizer的__init__:其中defaultdict:即使给了错误的索引,也会返回一个默认值,不会报错将导入的parameters放入param_groups的param中这一行利用param.grad对param进行更新import mathimport tor...原创 2019-12-12 16:40:50 · 1394 阅读 · 0 评论 -
SSD实现
prior box生成:对ground truth的编码:nms实现:test样本的Detect:原创 2019-12-03 16:35:39 · 257 阅读 · 0 评论 -
DenseNet实现
关于Dropout:每个denseblock中的小block后都有一个dropout,查看源码试图搞清楚这里dropout的实现方法,但没有查询到dropout的最终实现,在_VF中。为了减少GPU显存的使用,这里将第一个bn进行了checkpoint,即不储存反向传播的输出,在有需要时重新计算,这个操作以较小的时间开销换取了大量的显存空间。kaiming初始化实现:...原创 2019-12-02 11:08:16 · 218 阅读 · 0 评论 -
efficientnet实现
efficient内实现:forward:resnet中实现:原创 2019-12-01 15:16:09 · 925 阅读 · 0 评论