Skip to content

[FSDPv1] Only perform cat() during last microbatch backward() within FlattenParamsWrapper #751

[FSDPv1] Only perform cat() during last microbatch backward() within FlattenParamsWrapper

[FSDPv1] Only perform cat() during last microbatch backward() within FlattenParamsWrapper #751