Model save

README.md

@@ -1,7 +1,7 @@
 ---
 library_name: transformers
 license: apache-2.0
-base_model: microsoft/swin-base-patch4-window7-224
+base_model: itsLeen/vit-large-ai-or-not
 tags:
 - generated_from_trainer
 model-index:
@@ -14,15 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-large-ai-or-not
 
-This model is a fine-tuned version of [microsoft/swin-base-patch4-window7-224](https://huggingface.co/microsoft/swin-base-patch4-window7-224) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- eval_loss: 0.1679
-- eval_accuracy: 0.9801
-- eval_runtime: 50.6302
-- eval_samples_per_second: 73.553
-- eval_steps_per_second: 9.204
-- epoch: 7.9484
-- step: 7400
+This model is a fine-tuned version of [itsLeen/vit-large-ai-or-not](https://huggingface.co/itsLeen/vit-large-ai-or-not) on an unknown dataset.
 
 ## Model description
 
@@ -49,9 +41,13 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 16
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 30
+- num_epochs: 6
 - mixed_precision_training: Native AMP
 
+### Training results
+
+
+
 ### Framework versions
 
 - Transformers 4.44.2