Skip to content

Why use zero_module? #64

Open
Open

Description

@ZhangMingKun1

Thanks for your code for the project! It is a really nice work!

I am confused about why using zero_module, may lead to the zero_grad between the input and the output. It is possible to correctly train the model parameter with the expected grad?

Activity

  1. konpatp commented on Sep 28, 2023

    @konpatp
    Owner

    Is it true that zero module is the cause of zero grad? I'm not sure about this.

    By the way, we used zero grad module based on a previous work, but by itself, it also has a positive impact faster learning as well (as shown in the previous works).

  2. ZhangMingKun1 commented on Sep 29, 2023

    @ZhangMingKun1
    Author

    Thank you for your feedback! Could you please provide the paper title or GitHub link of the "previous works"?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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