src/sagemaker_defect_detection/classifier.py [38:43]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        num_classes: int,
        learning_rate: float,
        batch_size: int,
        momentum: float,
        weight_decay: float,
        seed: int,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/sagemaker_defect_detection/detector.py [42:47]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        num_classes: int,
        learning_rate: float,
        batch_size: int,
        momentum: float,
        weight_decay: float,
        seed: int,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -