fix bug of loger

cell_segmentation/inference/inference_cellvit_experiment_monuseg.py

@@ -92,7 +92,7 @@ class MoNuSegInference:
 
         # self.__instantiate_logger()
         self.__load_model()
-        self.__load_inference_transforms()
+        # self.__load_inference_transforms()
         self.__setup_amp()
         self.inference_dataset = MoNuSegDataset(
             dataset_path=dataset_path,
@@ -124,16 +124,14 @@ class MoNuSegInference:
 
     def __load_model(self) -> None:
         """Load model and checkpoint and load the state_dict"""
-        self.logger.info(f"Loading model: {self.model_path}")
 
         model_checkpoint = torch.load(self.model_path, map_location="cpu")
 
         # unpack checkpoint
         self.run_conf = unflatten_dict(model_checkpoint["config"], ".")
         self.model = self.__get_model(model_type=model_checkpoint["arch"])
-        self.logger.info(
-            self.model.load_state_dict(model_checkpoint["model_state_dict"])
-        )
+    
+        self.model.load_state_dict(model_checkpoint["model_state_dict"])
         self.model.eval()
         self.model.to(self.device)