7 Commits

Author SHA1 Message Date
Thalles Silva
727cbae12c Fixed loss and added pytorch built in mixed precision 2021-02-11 10:18:51 -03:00
Thalles Silva
55910e6107 fix loss function labels 2021-01-21 06:32:50 -03:00
Thalles Silva
13a7e646e8 fix loss function labels 2021-01-21 06:32:08 -03:00
Thalles Silva
6cd188ef68 Major refactor, small fixes 2021-01-18 07:33:12 -03:00
Thalles Silva
2c9536f731 Major refactor 2021-01-17 14:12:17 -03:00
Thalles
d1ff268e94 minor improvements 2020-03-14 07:01:49 -03:00
Thalles
45e3b3b7ef new pythonic implementation 2020-03-13 22:56:04 -03:00