
pytorch
华章致远
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
groups的简单文字理解
group这个参数确实是用做分组卷积的,但是现在用的比较多的是groups = in_channel,可以参考上面英文文档的最后一句。当groups = in_channel时,是在做的depth-wise conv的,具体思想可以参考MobileNet那篇论文。 ...原创 2019-12-31 09:36:07 · 674 阅读 · 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 · 250 阅读 · 0 评论