fix: fix problems with y architecture and checkpoint

2022-06-12 10:36:15 +02:00
parent fe729781fc
commit be7d7f43bd
4 changed files with 47 additions and 32 deletions
--- a/prototorch/y/library/init.py
+++ b/prototorch/y/library/init.py
@@ -1,5 +1,7 @@
 from .glvq import GLVQ
+from .gmlvq import GMLVQ

 __all__ = [
    "GLVQ",
+    "GMLVQ",
 ]