Update base_class.py

SforAiDl · Jun 11, 2021 · eca9657 · eca9657
1 parent 6dd4283
commit eca9657
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/KD_Lib/KD/__init__.py b/KD_Lib/KD/__init__.py
@@ -1,5 +1,4 @@
 from .vision import (
-    BaseClass,    
     VanillaKD,
     VirtualTeacher,
     SelfTraining,
@@ -19,3 +18,4 @@
 )
 
 from .text import BERT2LSTM, get_essentials
+from .common import BaseClass