Closed aneeshchauhan closed 5 years ago
Changes to model.py and trainer.py model.py: Fixed bug on usage of switching gradient computation on/off trainer.py: Fixed bug on handling 0 dimensional loss_values. This is an error I faced while running the tests
Changes to model.py and trainer.py model.py: Fixed bug on usage of switching gradient computation on/off trainer.py: Fixed bug on handling 0 dimensional loss_values. This is an error I faced while running the tests