Antti Tarvainen

icon company IPRally icon location Helsinki, Finland icon location Machine learning engineer

Results 1 issues of Antti Tarvainen

Fix support for models with one output (PyTorch implementation)

The PyTorch code currently crashes if the model has only one output. See comments on issue #7 .