pytorch-lightning
fix: skip weight copy for LayerNorm with elementwise_affine=False in Transformer Engine plugin
#21896

Closed

fix: skip weight copy for LayerNorm with elementwise_affine=False in Transformer Engine plugin #21896

xiaoyaoqilan wants to merge 1 commit into Lightning-AI:master from xiaoyaoqilan:fix/checkpoint-permission-error

fix: skip weight copy when LayerNorm has elementwise_affine=False (#2…

811aad8e

xiaoyaoqilan requested a review from

tchaton 17 days ago

xiaoyaoqilan requested a review from

justusschock 17 days ago

xiaoyaoqilan requested a review from

ethanwharris 17 days ago

xiaoyaoqilan closed this 5 days ago

Reviewers

tchaton

justusschock

ethanwharris

Assignees

No one assigned

Labels

None yet

Milestone

No milestone