(model_name)
| 35 | |
| 36 | |
| 37 | def is_vision_model(model_name): |
| 38 | if model_name in [ |
| 39 | # These models are also registered in examples/models |
| 40 | "dl3", |
| 41 | "edsr", |
| 42 | "mv2", |
| 43 | "mv3", |
| 44 | "vit", |
| 45 | "ic3", |
| 46 | "ic4", |
| 47 | "resnet18", |
| 48 | "resnet50", |
| 49 | # These models are not registered in examples/models but are available via |
| 50 | # torchvision |
| 51 | "convnext_small", |
| 52 | "densenet161", |
| 53 | "shufflenet_v2_x1_0", |
| 54 | ]: |
| 55 | return True |
| 56 | |
| 57 | return False |
| 58 | |
| 59 | |
| 60 | def get_vision_model_sample_input(): |
no outgoing calls
no test coverage detected