nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

fix train loss returns #32

Closed alphinside closed 4 years ago

alphinside commented 4 years ago

Type of changes

Please check the type of change your PR introduces: - [x] Bugfix - [ ] Feature - [ ] Code style update (formatting, renaming) - [ ] Refactoring (no functional changes, no api changes) - [ ] Build related changes - [ ] Documentation content changes - [ ] Other (please describe): ## What is the current behavior? Error on hypopt with train objective when using device 'cuda' - close #31 ## What is the new behavior?

Checklist