2020-11-23

最新推荐文章于 2024-01-21 10:31:19 发布

weixin_45908408

最新推荐文章于 2024-01-21 10:31:19 发布

阅读量1k

点赞数 1

分类专栏： pycharm python代码

本文链接：https://blog.csdn.net/weixin_45908408/article/details/110004343

版权

本文记录了在调试深度学习代码时遇到的TypeError问题，具体涉及upsample_bilinear2d函数的output_size参数类型错误。作者通过排除法和debug定位到问题在于批处理大小(batch size)的设置，当batch size不为1时，输出尺寸会被保存为Tensor而非所需的整数元组。解决方案是确保测试时batch size设为1。

摘要由CSDN通过智能技术生成

TypeError: upsample_bilinear2d(): argument ‘output_size’ must be tuple of ints, but found element of type Tensor at pos 1

调试代码遇到的大坑，报错如下：
在这里插入图片描述

D:\anconda\envs\python37\python.exe D:/document/memo/LDF-1117/train-coarse/test.py

Parameters...
datapath  : D:/date set/saliency_test/ECSSD
snapshot  : ./out/model-37
mode      : test
D:\anconda\envs\python37\lib\site-packages\torch\nn\functional.py:2506: 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.
  "See the documentation of nn.Upsample for details.".format(mode)