prototorch_models/prototorch/models
Alexander Engelsberger 7379c61966 [BUGFIX] Fix image visualization for some parameter combination
image visualization was broken if add_embeding was False, but data visualization was on.
2021-06-03 15:12:51 +02:00
..
__init__.py [BUGFIX] Early stopping example works now 2021-06-03 13:38:16 +02:00
abstract.py Remove unused import 2021-06-01 23:40:56 +02:00
callbacks.py [BUGFIX] Pruning example works on GPU now 2021-06-03 14:35:24 +02:00
cbc.py [WIP] Update CBC example 2021-06-02 00:45:33 +02:00
data.py Add MNIST datamodule and training mixin factory. 2021-05-28 16:33:31 +02:00
glvq.py [BUGFIX] Pruning example works on GPU now 2021-06-03 14:35:24 +02:00
lvq.py Refactor non-gradient-lvq models into lvq.py 2021-05-25 20:37:34 +02:00
probabilistic.py [BUGFIX] Probabilistic Models work on GPU now 2021-06-03 14:05:44 +02:00
unsupervised.py Tweak repr 2021-06-02 01:07:48 +02:00
vis.py [BUGFIX] Fix image visualization for some parameter combination 2021-06-03 15:12:51 +02:00