(name: str, num_classes: int, **kwargs)
| 12 | |
| 13 | |
| 14 | def get_network(name: str, num_classes: int, **kwargs) -> Module | None: |
| 15 | return \ |
| 16 | resnet18( |
| 17 | num_classes=num_classes) if name == 'resnet18' else\ |
| 18 | resnet34( |
| 19 | num_classes=num_classes) if name == 'resnet34' else\ |
| 20 | resnet50( |
| 21 | num_classes=num_classes) if name == 'resnet50' else\ |
| 22 | resnet101( |
| 23 | num_classes=num_classes) if name == 'resnet101' else\ |
| 24 | resnet152( |
| 25 | num_classes=num_classes) if name == 'resnet152' else\ |
| 26 | resnet18_cifar( |
| 27 | num_classes=num_classes) if name == 'resnet18Cifar' else\ |
| 28 | resnet34_cifar( |
| 29 | num_classes=num_classes) if name == 'resnet34Cifar' else\ |
| 30 | resnet50_cifar( |
| 31 | num_classes=num_classes) if name == 'resnet50Cifar' else\ |
| 32 | resnet101_cifar( |
| 33 | num_classes=num_classes) if name == 'resnet101Cifar' else \ |
| 34 | resnet152_cifar( |
| 35 | num_classes=num_classes) if name == 'resnet152Cifar' else \ |
| 36 | densenet121( |
| 37 | num_classes=num_classes) if name == 'densenet121' else\ |
| 38 | densenet161( |
| 39 | num_classes=num_classes) if name == 'densenet161' else\ |
| 40 | densenet169( |
| 41 | num_classes=num_classes) if name == 'densenet169' else\ |
| 42 | densenet201( |
| 43 | num_classes=num_classes) if name == 'densenet201' else \ |
| 44 | densenet121_cifar( |
| 45 | num_classes=num_classes) if name == 'densenet121Cifar' else \ |
| 46 | densenet161_cifar( |
| 47 | num_classes=num_classes) if name == 'densenet161Cifar' else \ |
| 48 | densenet169_cifar( |
| 49 | num_classes=num_classes) if name == 'densenet169Cifar' else \ |
| 50 | densenet201_cifar( |
| 51 | num_classes=num_classes) if name == 'densenet201Cifar' else\ |
| 52 | swin_t( |
| 53 | num_classes=num_classes, **kwargs) if name == 'swin_t' else\ |
| 54 | swin_s( |
| 55 | num_classes=num_classes, **kwargs) if name == 'swin_s' else\ |
| 56 | swin_b( |
| 57 | num_classes=num_classes, **kwargs) if name == 'swin_b' else\ |
| 58 | swin_l( |
| 59 | num_classes=num_classes, **kwargs) if name == 'swin_l' else \ |
| 60 | cct( |
| 61 | num_classes=num_classes, **kwargs) if name == 'cct' else \ |
| 62 | FocalNet( |
| 63 | num_classes=num_classes, **kwargs) if name == 'focalnet' else \ |
| 64 | mobilenetv3( |
| 65 | num_classes=num_classes, **kwargs) if name == 'mobilenetv3' else \ |
| 66 | None |
no test coverage detected