lgwang

Hello, have you solved the problem that the prediction results are all black during prediction? Looking forward to your answer

> 感谢您的解决方案。在 meth：torch.autocast 中设置 'dtype=torch.bfloat16' 有助于解决 'NAN' 丢失和黑色 predition 的问题。 Hi, could you please tell me which folder it is in?

> > > 感谢您的解决方案。在 meth：torch.autocast 中设置 'dtype=torch.bfloat16' 有助于解决 'NAN' 丢失和黑色 predition 的问题。您好，您能告诉我它在哪个文件夹中吗？ > > 之前使用 torch.autocast（device.type if device.type ！= 'mps' else 'cpu'， enabled=amp）train.py： > > 之后使用 torch.autocast（device.type if device.type ！=...