updates

buildingamind · Dec 9, 2023 · 5fcab6e · 5fcab6e
1 parent 528bfbe
commit 5fcab6e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -45,7 +45,8 @@ ViT-CoT
 └── scripts: directory containing scripts to train and test the model
 │ └── train_vit.sh
 │ ├── train_vit.py - contains model training loop
-│ ├── evaluate.py - contains encoder+linear probe evaluation code
+│ ├── evaluate.py - contains encoder+linear probe training loop
+│ ├── evaluator.py - contains encoder+linear probe training logic
 ├── media: directory containing images and videos for the readme
 ```