pytorch
cefc4d5c - Fix DDP incompatibility issue with nn.MultiheadAttention.

Commit

5 years ago

Fix DDP incompatibility issue with nn.MultiheadAttention.

References

Author

Guanheng Zhang

Parents

Loading