(use_pretrained, exportable=False)
| 76 | |
| 77 | |
| 78 | def _make_pretrained_efficientnet_lite3(use_pretrained, exportable=False): |
| 79 | efficientnet = torch.hub.load( |
| 80 | "rwightman/gen-efficientnet-pytorch", |
| 81 | "tf_efficientnet_lite3", |
| 82 | pretrained=use_pretrained, |
| 83 | exportable=exportable |
| 84 | ) |
| 85 | return _make_efficientnet_backbone(efficientnet) |
| 86 | |
| 87 | |
| 88 | def _make_efficientnet_backbone(effnet): |
no test coverage detected