| PARAMETERS: | |
| img_resize: 256 | |
| img_crop: 256 | |
| num_workers: 72 | |
| device: "cpu" | |
| dataset: "Artificio/WikiArt_mini_demos" | |
| metric: "L2" | |
| ROBUST: | |
| arch: "resnet50" | |
| arch_path: "models/robust_resnet50.pt" | |
| load_from: "torchvision" | |
| layer: "avgpool" | |
| embedding_col: "resnet50_robust_features_2048" | |
| NONROBUST: | |
| arch: "resnet50" | |
| arch_path : "" | |
| load_from: "timm" | |
| layer: "global_pool" | |
| embedding_col: "resnet50_non_robust_features_2048" | |