How to add pixel-level loss #166

Open

opened

on Dec 27, 2024

How can I add pixel-level loss in stage2, such as the MSE of the entire image? I see that the loss in the code is just diffusion noise. How can I change it to the MSE of the entire image?
Thanks！

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Type

No type

Fields

No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests