Skip to content

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

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

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

Annotations

2 errors

pre-commit (3.8.12)

cancelled May 13, 2024 in 8s