Fix name error for USE_FED_LOSS

python_utils/get_model.py

@@ -10,18 +10,9 @@ def get_set_up():
 
 def load_model():
     # return None
-    # def predictor(img):
-    #     return {}
-    # return predictor
-    # import some common detectron2 utilities
     import torch
-    from detectron2 import model_zoo
     from detectron2.engine import DefaultPredictor
     from detectron2.config import get_cfg
-    from detectron2.data.datasets import register_coco_instances
-
-    import os
-    import numpy as np
 
     ## define relevant parameters
     cfg = get_cfg()
@@ -37,7 +28,7 @@ def load_model():
     ## some messages will appear when using the model that certain weights are not being used 
     ## but these are used during training and not inference and shouldn't affect the model performance
     ## code below
-    cfg.MODEL.ROI_BOX_HEAD.USE_FED_LOSS = false
+    cfg.MODEL.ROI_BOX_HEAD.USE_FED_LOSS = False
     
     predictor = DefaultPredictor(cfg)