Closed aneeshchauhan closed 5 years ago
bug removed in gradient switching using pytorch 0.4 (models.py); handling 0dim arrays in trainer.py
This is great, thanks Aneesh!
bug removed in gradient switching using pytorch 0.4 (models.py); handling 0dim arrays in trainer.py