Zimo He comments

Results 2 comments of


                                            Zimo He

train.py里面 ‘’‘optimizer.zero_grad() x_0 = images.to(device) loss = trainer(x_0).sum() / 1000. loss.backward()’‘’ 就是在训练U-Net啊

I think maybe you can modify the p_losses function in LatentDiffusion class.