diff --git a/dinov2/dinov2/configs/eval/vitb14_pretrain.yaml b/dinov2/dinov2/configs/eval/vitb14_pretrain.yaml new file mode 100644 index 0000000..117d0f0 --- /dev/null +++ b/dinov2/dinov2/configs/eval/vitb14_pretrain.yaml @@ -0,0 +1,6 @@ +student: + arch: vit_base + patch_size: 14 +crops: + global_crops_size: 518 # this is to set up the position embeddings properly + local_crops_size: 98 \ No newline at end of file diff --git a/dinov2/dinov2/configs/eval/vitg14_pretrain.yaml b/dinov2/dinov2/configs/eval/vitg14_pretrain.yaml new file mode 100644 index 0000000..a96dd5b --- /dev/null +++ b/dinov2/dinov2/configs/eval/vitg14_pretrain.yaml @@ -0,0 +1,7 @@ +student: + arch: vit_giant2 + patch_size: 14 + ffn_layer: swiglufused +crops: + global_crops_size: 518 # this is to set up the position embeddings properly + local_crops_size: 98 \ No newline at end of file diff --git a/dinov2/dinov2/configs/eval/vitl14_pretrain.yaml b/dinov2/dinov2/configs/eval/vitl14_pretrain.yaml new file mode 100644 index 0000000..7a98454 --- /dev/null +++ b/dinov2/dinov2/configs/eval/vitl14_pretrain.yaml @@ -0,0 +1,6 @@ +student: + arch: vit_large + patch_size: 14 +crops: + global_crops_size: 518 # this is to set up the position embeddings properly + local_crops_size: 98 \ No newline at end of file diff --git a/dinov2/dinov2/configs/eval/vits14_pretrain.yaml b/dinov2/dinov2/configs/eval/vits14_pretrain.yaml new file mode 100644 index 0000000..afbdb4b --- /dev/null +++ b/dinov2/dinov2/configs/eval/vits14_pretrain.yaml @@ -0,0 +1,6 @@ +student: + arch: vit_small + patch_size: 14 +crops: + global_crops_size: 518 # this is to set up the position embeddings properly + local_crops_size: 98 \ No newline at end of file