yolop 训练模型错误

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

img

这个错误可能是因为在计算过程中,某个需要整数类型(int64)的变量被赋予了一个浮点数(float)类型的值。
为了找到并解决问题,您需要定位到引起错误的代码部分。

首先,需要查看错误消息中的堆栈跟踪以获取有关错误发生位置的详细信息。您可以在训练过程中的日志输出中找到这些信息。堆栈跟踪将包含文件名、行号和函数名,这将有助于您确定错误的来源。

一旦您找到了出错的代码行,请检查该行中涉及的变量和操作。请确保您在需要整数类型的地方使用整数类型的值。有时,可能需要将浮点数转换为整数。
可以使用以下方法将浮点数转换为整数:

int_value = int(float_value)