()
| 130 | f_n.write(s + '\n') |
| 131 | |
| 132 | def test(): |
| 133 | args = get_args() |
| 134 | imagenet_r, imagenetr_labels = load_data(DATA_FOLDER[args.dataset]) |
| 135 | model, processor = load_model(CLIP_MODELS[args.model]) |
| 136 | res, acc = classify_imagenetr( |
| 137 | imagenet_r, imagenetr_labels, model, processor, device='cuda') |
| 138 | res = np.array(res) |
| 139 | np.savetxt('res.txt', res, fmt='%d') |
| 140 | print(acc) |
| 141 | |
| 142 | def sweep(): |
| 143 | # Gives all results from all datasets across all models |
no test coverage detected