博主这个问题是在训练CenterNet时遇到的,解决方法为使得chunk_size和自己可用的GPU数匹配。剩下的就是调整batch_size了。
RuntimeError: given chunk sizes don‘t sum up to the tensor‘s size (sum(chunk_sizes) == 48, but expec
原创
©著作权归作者所有:来自51CTO博客作者wx5ba0c87f1984b的原创作品,请联系作者获取转载授权,否则将追究法律责任
提问和评论都可以,用心的回复会被更多人看到
评论
发布评论
相关文章
-
Linux 文件管理命令 md5sum awk
Linux 文件管理命令md5sum:MD5 函数值计算和检查 awk:模式匹配语言
Linux命令 md5sum awk