Skip to content

An error in Colab: Tutorial: Rectified Flow with Neural Network.ipynb #28

@chwbob

Description

@chwbob

When training rectified_flow_2, the code rectified_flow_2.net = copy.deepcopy(rectified_flow_1) is wrong and useless. In fact, I think you should use rectified_flow_2.model.load_state_dict(rectified_flow_1.model.state_dict()) to load pretrained parameters of rectified_flow_1.

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