Go
pytorch
f34c6f78
- [functorch] Use the smallest batch size we can in vmap testing (pytorch/functorch#936)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
2 years ago
[functorch] Use the smallest batch size we can in vmap testing (pytorch/functorch#936) This speeds up test_vmap by 16% real time on my machine
Author
zou3519
Committer
bigfootjon
Parents
0d9fd3d5
Loading