Closed pavlin-policar closed 4 years ago
@dkobak Please check that this indeed works as intended. I quickly tested it on mnist, and it seems to work ok.
I haven't had a chance to do actual testing, but I looked through the code and everything looks fine. I think you can go ahead and merge.
Description of changes
Fixes #131
Includes