博主这个问题是在训练CenterNet时遇到的,解决方法为使得chunk_size和自己可用的GPU数匹配。剩下的就是调整batch_size了。
QQ:战、血舞皇朝、2125364717
RuntimeError: given chunk sizes don‘t sum up to the tensor‘s size (sum(chunk_sizes) == 48, but expec
阅读 33
2022-02-26
博主这个问题是在训练CenterNet时遇到的,解决方法为使得chunk_size和自己可用的GPU数匹配。剩下的就是调整batch_size了。
QQ:战、血舞皇朝、2125364717
相关推荐
精彩评论(0)