![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
华章致远
这个作者很懒,什么都没留下…
展开
-
groups的简单文字理解
group这个参数确实是用做分组卷积的,但是现在用的比较多的是groups = in_channel,可以参考上面英文文档的最后一句。当groups = in_channel时,是在做的depth-wise conv的,具体思想可以参考MobileNet那篇论文。 ...原创 2019-12-31 09:36:07 · 633 阅读 · 0 评论 -
torch中的max
torch中max的各种用法 torch.max()[0], 只返回最大值的每个数 troch.max()[1], 只返回最大值的每个索引 torch.max()[1].data 只返回variable中的数据部分(去掉Variable containing:) torch.max()[1].data.numpy() 把数据转化成numpy ndarry torch.max()[1].data.n...转载 2019-05-11 10:47:18 · 200 阅读 · 0 评论