超级会员免费看
本文探讨了在PyTorch训练中遇到的关于bilinear上采样方式的警告，解释了从PyTorch 0.4.0开始align_corners默认为False的变化。同时，分析了TensorRT与PyTorch在align_corners设置不一致导致的问题，指出当TensorRT的mode=bilinear时，align_corners需设为True以匹配PyTorch的行为。还引用了ONNX官方文档对上采样模式的详细说明。
问题1：pytorch训练过程中遇到：
UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details.
提示的意思是：自从pytorch0.4.0以后，upsample中当选择bilinear[双线性差值]方式时，align_corners将默认设置为False。如果有需求，可自行将其设置为True。这是正常的warning而已，不用担心。
问题2：TensorRT和Pytorch中关于align_corners设置不一致的问题
bilinear IResizeLayer setting align_corners=False makes different result to torch interpolate为了详细的了解这个问题并解决该问题，查了一些资料，记录下来
（1）upsampling参数：
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
