fix: wrong module import

archinetai · Jul 19, 2022 · e648ab4 · e648ab4
1 parent 6954971
commit e648ab4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vat_pytorch/__init__.py b/vat_pytorch/__init__.py
@@ -2,4 +2,4 @@
 from .alice import ALICELoss
 from .alicepp import ALICEPPLoss, ALICEPPModule
 from .utils import kl_loss, sym_kl_loss, js_loss, inf_norm, default
-from .modules import *
+from .models import *