Fix typo in CLPI model

TuanBC · Oct 27, 2022 · 9c925cc · 9c925cc
1 parent dbd61cd
commit 9c925cc
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/CV/CLPI-Collaborative-Learning-for-Diabetic-Retinopathy-Grading/clfnet/__init__.py b/CV/CLPI-Collaborative-Learning-for-Diabetic-Retinopathy-Grading/clfnet/__init__.py
@@ -29,8 +29,6 @@
 from . import densenet
 from . import inception_v4 as inception
 
-from learning_rate import cosine_decay_with_warmup
-
 __all__ = ["create_model"]
 
 ResNetModels = sorted(name for name in resnet.__dict__
@@ -197,4 +195,4 @@ def net(self, class_dim=5, CAM=False):
             cam_heatmap = self.create_cam_op(predict, class_dim, heatmaps)
             return [loss, accuracy, predict, hacked_img_id, cam_heatmap], reader
 
-        return [loss, accuracy, predict, hacked_img_id], reader
+        return [loss, accuracy, predict, hacked_img_id], reader