首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Resnet18第一层输出尺寸

Resnet18第一层输出尺寸
EN

Stack Overflow用户
提问于 2020-06-09 22:24:32
回答 1查看 1.3K关注 0票数 0

我正在研究PyTorch中的模型实现。第一层为卷积层,滤波器大小为7,步长为2,pad = 3。网络的标准输入尺寸为224x224x3。基于这些数字,输出维数为(224 +3*2-7)/2+ 1,它不是整数。原始实现是否包含非整数维?我看到网络在FC层之前有自适应池,所以变量输入维度不是问题(我通过改变输入大小来测试这一点)。我是不是做错了什么,或者为什么作者在设计ResNet时选择了一个非整数维?

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2020-06-09 22:47:43

维度总是必须是整数。来自nn.Conv2d - Shape

只向底部关闭的括号表示地板操作(向下)。计算结果如下:

代码语言:javascript
运行
复制
import math

math.floor((224 + 3*2 - 7)/2 + 1) # => 112

# Or using the integer division (two slashes //)
(224 + 3*2 - 7) // 2 + 1 # => 112

使用整数除法具有相同的效果,因为这总是将其舍入到最近的整数。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/62292854

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档