Update README.md

README.md

@@ -22,7 +22,7 @@ This is a fine-tuned deit_small_patch16_224 model trained on Pinterest Face Reco
 
 ## Training Details
 
-- Number of prototypes: 2000
+- Number of prototypes: 1550
 - Prototype size: 1×1
 - Training process: Warm up → Joint training → Push → Last layer fine-tuning
 - Weight coefficients:
@@ -32,9 +32,6 @@ This is a fine-tuned deit_small_patch16_224 model trained on Pinterest Face Reco
   - L1: 0.01
   - Orthogonal: 0.001
   - Coherence: 0.003
-- Training set size: 70420
-- Push set size: 13979
-- Test set size: 3555
 - Batch size: 128
 
 ## Dataset Description
@@ -63,4 +60,6 @@ predicted_label = outputs.logits.argmax(-1).item()
 
 ## Additional Information
 
-For more details about the implementation and training process, please visit the [GitHub repository](https://github.com/ayushnangia/ProtoViT).
+Github repo by authors of the paper [GitHub repository][https://github.com/Henrymachiyu/ProtoViT]
+
+For more details about the implementation and training process, please visit the my fork of ProtoVit [GitHub repository](https://github.com/ayushnangia/ProtoViT).