diffusion loss

[05Pikachu24]

佬，你论文的公式11，扩散loss体现在哪里啊

[Melmaphother]

感谢你的关注！
loss 在 https://github.com/Melmaphother/TimeDART/blob/main/exp/exp_timedart.py 第 68 行：

def _select_criterion(self):
if self.args.task_name == "finetune" and self.args.downstream_task == "classification":
criterion = nn.CrossEntropyLoss()
print("Using CrossEntropyLoss")
else:
criterion = nn.MSELoss()
print("Using MSELoss")
return criterion

形式上 diff loss 就是 L2 loss 的形式，这里用 MSE 或者 L2 loss 都可以。