数据类型错误

**描述错误** 使用 train.py 训练模型时，存在与数据类型相关的错误。它既可以输入为浮点数或张量，也可以将 [0,0.99] 视为整数和浮点数，并导致错误。