added eval_knn.py

clemsgrs · Mar 7, 2024 · 6f17006 · 6f17006
1 parent 8d40180
commit 6f17006
Show file tree

Hide file tree

Showing 2 changed files with 476 additions and 0 deletions.
diff --git a/dino/config/knn.yaml b/dino/config/knn.yaml
@@ -0,0 +1,35 @@
+data:
+  features_dir:
+  query_csv: ''
+  test_csv: ''
+
+output_dir: 'output'
+experiment_name: 'eval'
+
+batch_size_per_gpu: 256
+
+nb_knn: 20
+temperature: 0.07
+save_features: false
+label_name: ''
+
+model:
+  arch: vit_small
+  patch_size: 16
+  pretrained_weights: ''
+  checkpoint_key: 'teacher'
+
+speed:
+  use_cuda: true
+  num_workers: 16
+
+wandb:
+  enable: true
+  project: 'vision'
+  username: 'vlfm'
+  exp_name: 'eval'
+  tags: ['${experiment_name}', 'knn', '${student.arch}']
+  dir: '/home/user'
+  group:
+  to_log:
+  resume_id: