(modelname)
| 71 | |
| 72 | |
| 73 | def load_model(modelname): |
| 74 | device = "cuda" if torch.cuda.is_available() else "cpu" |
| 75 | model, preprocess = clip.load(modelname, device) |
| 76 | return model, preprocess |
| 77 | |
| 78 | |
| 79 | def classify_imagenetr(imagenet_r, imagenetr_labels, model, preprocess, device): |
no test coverage detected