neodroidvision.classification.architectures.resnet_retrain.resnet_retrain¶
- neodroidvision.classification.architectures.resnet_retrain.resnet_retrain(num_classes: int, freeze_first_num: int = 6, pretrained: bool = True, resnet_factory: callable = <function resnet18>) Tuple[ResNet, List[Parameter]][source]¶
- Parameters
num_classes –
freeze_first_num –
pretrained –
resnet_factory –
Returns:
