Loss of editing?

[tingxueronghua]

I implement the editing loss described in your paper and overfit the MagicBrush dataset,
However I found that the loss are very large and the generated image quickly becomes noises. Is there any tricks required?

[staoxiao]

Hi, @tingxueronghua , you can refer to https://github.com/VectorSpaceLab/OmniGen/blob/main/train.py#L208-L228 for detailed settings.