benchmark
d0a0ced7
- Minor improvements to check_device
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 years ago
Minor improvements to check_device Change Exception to RuntimeError. Update check inputs to a recursive implementation.
References
#443 - Add test to verify models and inputs are using the expected device
Author
aaronenyeshi
Parents
d8dc7d8b
Loading