in src/sagemaker_defect_detection/detector.py [0:0]
def __init__(
self,
train_rpn: bool,
train_roi: bool,
finetune_rpn: bool,
finetune_roi: bool,
data_path: str,
backbone: str,
num_classes: int,
learning_rate: float,
batch_size: int,
momentum: float,
weight_decay: float,
seed: int,
pretrained_mfn_ckpt: Optional[str] = None,
pretrained_rpn_ckpt: Optional[str] = None,
pretrained_roi_ckpt: Optional[str] = None,
finetuned_rpn_ckpt: Optional[str] = None,
finetuned_roi_ckpt: Optional[str] = None,
resume_sagemaker_from_checkpoint: Optional[str] = None,
**kwargs,