关于 loss 计算代码

[sunjian2015]

RealDPO/finetune/models/cogvideox_i2v/real_dpo_trainer.py

Line 284 in ee15396

lose_loss = torch.mean((weights * (lose_latent_pred - lose_latent) ** 2).reshape(batch_size, -1), dim=1)

这里的 lose_latent 是写错了吗？不应该是 lose_latent[n_idx] 吗？

[nini0919]

感谢你细心的查阅我们的论文和代码，开源代码上确实是写错了，这个在开源整理代码规范时，遗漏了索引。我们将重新修正，谢谢指出问题！

[sunjian2015]

您好，这个参数代码里是固定为 5000，我改成 wan2.1-i2v-1.3b 尝试训了一下，发现 loss 波动特别大，跟这个有关系吗？我看您训练脚本里这个参数是 1 呢

RealDPO/finetune/models/cogvideox_i2v/real_dpo_trainer.py

Line 287 in 44b5c3b

beta_dpo = 5000