Model save

README.md

@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.5
+      value: 0.8333333333333334
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -33,12 +33,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [microsoft/swin-base-patch4-window7-224-in22k](https://huggingface.co/microsoft/swin-base-patch4-window7-224-in22k) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.0945
-- Accuracy: 0.5
-- Auc Overall: 0.5833
-- Auc Class 0: 0.5
+- Loss: 0.7837
+- Accuracy: 0.8333
+- Auc Overall: 0.875
+- Auc Class 0: 1.0
 - Auc Class 1: 0.625
-- Auc Class 2: 0.625
+- Auc Class 2: 1.0
 
 ## Model description
 
@@ -66,7 +66,7 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 7
+- num_epochs: 3
 
 ### Training results
 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a4b0327ec1c27065292a064257d9aaca3c5cdbbd3cfaded2515773015fe76622
+oid sha256:bedc598d44f8d7a3676004ffc20a543e7e751c4cdb3b4aeec636bcca360d6a2a
 size 347502916

runs/Sep04_10-43-53_data-science-wbi/events.out.tfevents.1725446635.data-science-wbi

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e26d8905fe1e98f73246089f11cf3279e0b2315369479864a22eecb3e89b7713
-size 6025
+oid sha256:ef302f96d9c7bf98cac57f40a574878660b02042e1409f230398cd0b98e4486b
+size 9164