YOLOPOSE训练问题

训练YOLOPOSE过程中的问题

您好!想请教您一些训练yolopose的问题,我在服务器上训练的时候会遇到报错


RuntimeError: result type Float can't be cast to the desired output type long int

百度后修改了loss.py里的
替换成:anchors, shape = self.anchors[i], p[i].shape

替换成:anchors, shape = self.anchors[i], p[i].shape

修改完可以成果训练,但是每次训练后的验证标签和损失都是0.训练了十几个epoch都是这种问题,想请教您是如何成功跑通的呢?感激不尽!