Custom non-gradient training

2021-05-18 19:49:16 +02:00
parent 246719b837
commit eefec19c9b
3 changed files with 101 additions and 18 deletions
--- a/prototorch/models/init.py
+++ b/prototorch/models/init.py
@@ -1,8 +1,8 @@
 from importlib.metadata import PackageNotFoundError, version

 from .cbc import CBC
-from .glvq import (GLVQ, GMLVQ, GRLVQ, LVQ1, LVQ21, LVQMLN, ImageGLVQ,
-                   ImageGMLVQ, SiameseGLVQ)
+from .glvq import (GLVQ, GMLVQ, GRLVQ, GLVQ1, GLVQ21, LVQ1, LVQ21, LVQMLN,
+                   ImageGLVQ, ImageGMLVQ, SiameseGLVQ)
 from .knn import KNN
 from .neural_gas import NeuralGas
 from .vis import *