File size: 480 Bytes
068d302 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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"
|