prototorch_models/prototorch/models/callbacks
Jensun Ravichandran fef73e2fbf [BUG] NaN when training with selection initializer
How to reproduce:
Run the `glvq_spiral.py` file under `examples/`.

The error seems to occur when using a lot of prototypes in combination with the
`StratifiedSelectionInitializer`. Using only a prototype per class, or using
another initializer like the `StratifiedMeanInitializer` seems to make the
problem go away.
2021-04-29 19:09:10 +02:00
..
visualization.py [BUG] NaN when training with selection initializer 2021-04-29 19:09:10 +02:00