Add backwards compatible import

lightly-ai · Dec 11, 2023 · bc8adad · bc8adad
1 parent 3043306
commit bc8adad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lightly/loss/memory_bank.py b/lightly/loss/memory_bank.py
@@ -0,0 +1,2 @@
+# For backwards compatibility as memory_bank module was previously in loss module.
+from lightly.models.modules.memory_bank import MemoryBankModule