src/sagemaker_defect_detection/detector.py [113:119]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            if self.finetuned_rpn_ckpt and self.finetuned_roi_ckpt:  # extra finetune rpn
                self.mfn = load_checkpoint(
                    Classification(self.backbone, self.num_classes - 1).mfn, self.finetuned_rpn_ckpt, prefix="mfn"
                )
                freeze(self.mfn)

                self.rpn = load_checkpoint(RPN(), self.finetuned_rpn_ckpt, prefix="rpn")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/sagemaker_defect_detection/detector.py [140:146]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            if self.finetuned_rpn_ckpt and self.finetuned_roi_ckpt:  # extra finetune roi
                self.mfn = load_checkpoint(
                    Classification(self.backbone, self.num_classes - 1).mfn, self.finetuned_rpn_ckpt, prefix="mfn"
                )
                freeze(self.mfn)

                self.rpn = load_checkpoint(RPN(), self.finetuned_rpn_ckpt, prefix="rpn")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



