微信扫一扫
博主这个问题是在训练CenterNet时遇到的,解决方法为使得chunk_size和自己可用的GPU数匹配。剩下的就是调整batch_size了。
QQ:战、血舞皇朝、2125364717