Model save

README.md

@@ -33,7 +33,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/vit-large-patch16-224](https://huggingface.co/google/vit-large-patch16-224) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1290
+- Loss: 0.0332
 - Accuracy: 1.0
 
 ## Model description
@@ -60,7 +60,7 @@ The following hyperparameters were used during training:
 - gradient_accumulation_steps: 4
 - total_train_batch_size: 32
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: cosine_with_restarts
+- lr_scheduler_type: linear
 - num_epochs: 15
 - mixed_precision_training: Native AMP
 
@@ -68,7 +68,7 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| 0.2379        | 6.6667 | 10   | 0.1290          | 1.0      |
+| 0.056         | 6.6667 | 10   | 0.0332          | 1.0      |
 
 
 ### Framework versions

runs/Nov12_16-08-39_5ee1c52bb489/events.out.tfevents.1731427733.5ee1c52bb489.1573.5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:87169f6445982ce3b7116ba14b835127f2681256a21b5be2bf2cfad6b0870a65
-size 5643
+oid sha256:d5b3775f33b6996945607b05a8735efc609790ffc0b6a4d8abd4075363328769
+size 5991