[FEATURE] Add PruneLoserPrototypes Callback

2021-06-02 03:52:41 +02:00
parent a3f5d7d113
commit 8851d1bbc9
2 changed files with 43 additions and 0 deletions
--- a/prototorch/models/init.py
+++ b/prototorch/models/init.py
@@ -2,6 +2,7 @@

 from importlib.metadata import PackageNotFoundError, version

+from .callbacks import PruneLoserPrototypes
 from .cbc import CBC, ImageCBC
 from .glvq import (
    GLVQ,