Skip to content

Question about train code #5

@terry2100

Description

@terry2100

Thanks for your work!
In the code for image2image_loss, why set vit_embeds.requires_grad = True?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions