如果不预测 xstart,则无法拟合模型

非常感谢您分享这篇伟大的作品。 我尝试使用 DiT 处理我的数据集,但当我使用 predict_xstart = False 来训练模型时,遇到了一些问题。损失略有下降,似乎停滞在一个高水平上,不会再下降,而当我使用 predict_xstart = True 时,损失很快就下降了。