RuntimeError: "nll_loss_out_frame" not implemented for 'Half'

[Li-Jicheng]

跑https://colab.research.google.com/github/shibing624/MedicalGPT/blob/main/run_training_dpo_pipeline.ipynb#scrollTo=J5kYehpzESyt （run_training_dpo_pipeline.ipynb）这个脚本时，pretrain阶段报错

[shibing624]

去掉--fp16, 设置 --torch_dtype auto

[Li-Jicheng]

去掉--fp16, 设置 --torch_dtype auto

感谢回复，操作后依然报错