Skip to content

Commit

Permalink
Fix api intg test
Browse files Browse the repository at this point in the history
  • Loading branch information
jaegukhyun committed Apr 8, 2024
1 parent 079fd70 commit 2c54c43
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions src/otx/algo/detection/ssd.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 2c54c43

Please sign in to comment.