3 Commits

Author SHA1 Message Date
Thalles
542b915e6e fixed bug missing zero_grad() 2020-03-01 10:19:41 -03:00
Thalles
88dcdf6d06 support to use other pytorch datasets 2020-02-29 19:32:37 -03:00
Thalles
5ed8a2e453 added tensorboard support 2020-02-24 18:23:44 -03:00