From 2c54c43328a147cc35945064e61c0fbaeb9eb2cd Mon Sep 17 00:00:00 2001 From: jaegukhyun Date: Mon, 8 Apr 2024 13:02:56 +0900 Subject: [PATCH] Fix api intg test --- src/otx/algo/detection/ssd.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/otx/algo/detection/ssd.py b/src/otx/algo/detection/ssd.py index f2b1d9df8f6..3f136bb0270 100644 --- a/src/otx/algo/detection/ssd.py +++ b/src/otx/algo/detection/ssd.py @@ -375,9 +375,10 @@ def device(self) -> device: else: return buf.device - self.classification_layers = self.get_classification_layers(self.config, "model.") - self.config.pop("type") - detector = SingleStageDetector(**convert_conf_to_mmconfig_dict(self.config)) + config = deepcopy(self.config) + self.classification_layers = self.get_classification_layers(config, "model.") + config.pop("type") + detector = SingleStageDetector(**convert_conf_to_mmconfig_dict(config)) if self.load_from is not None: load_checkpoint(detector, self.load_from, map_location="cpu") return detector