对于交叉熵损失，模型的输出类型需要为torch.float32,而标签的类型也需要为torch.float32