src/transformers/trainer.py · fb0a38b4f275727d6228fb4a78c15c6dd8480e91 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

Move torch.compile() wrapping after DDP/FSDP wrapping to ensure correct graph... · fb0a38b4
Antoni Viros authored Mar 20, 2023
```
Move torch.compile() wrapping after DDP/FSDP wrapping to ensure correct graph breaks during training (#22279)
```
fb0a38b4

trainer.py 181 KB