fix #1

SforAiDl · Jul 4, 2021 · 4d12fdb · 4d12fdb
1 parent 7e76967
commit 4d12fdb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/KD_Lib/KD/common/base_class.py b/KD_Lib/KD/common/base_class.py
@@ -56,7 +56,7 @@ def __init__(
         if device == "cpu":
             self.device = torch.device("cpu")
         elif device == "cuda":
-	    if torch.cuda.is_available():
+            if torch.cuda.is_available():
                 self.device = torch.device("cuda")
             else:
                 print(

diff --git a/tests/test_KD_Lib.py b/tests/test_KD_Lib.py
@@ -23,7 +23,7 @@
     ProbShift,
     LabelSmoothReg,
     DML,
-    BaseClass
+    BaseClass,
 )
 
 from KD_Lib.models import (
@@ -164,7 +164,8 @@ def test_LSTMNet():
 #   Strategy TESTS
 #
 
-def test_BaseClass()
+
+def test_BaseClass():
     teac = Shallow(hidden_size=400)
     stud = Shallow(hidden_size=100)