EleutherAI/oslo

TODO : Zero stage 1,2 backward

hyeinhyun opened this issue · 0 comments

hyeinhyun commented

Describe a TODO feature

  • Change Zero logic about backward function
    as-is : optimizer.backward
    to-be : loss.backward

Assignees

  • hyein.hyun