A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.42k
stars
1.4k
forks
source link
Reformat Grad Output If It's Not Channels Last #1837
Closed
alpha0422 closed 2 months ago