Mitchins
/

image-medium-classifier-efficientnetv2-s-v1

Image Classification

tf_efficientnetv2_s

anime-classification

rendered-graphics

Model card Files Files and versions

Mitchins commited on about 1 month ago

Commit

0dbf29b

·

verified ·

1 Parent(s): 5d2a96b

Upload README.md with huggingface_hub

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -94,7 +94,7 @@ print(f"{labels[pred_class]}: {probs[0, pred_class]:.2%}")
 ## Training Details
-- **Augmentation:** Raw (resize only)
 - **Optimizer:** AdamW (lr=0.001)
 - **Loss:** CrossEntropyLoss with class weighting
 - **Epochs:** 20

 ## Training Details
+- **Augmentation:** None (raw resize to 224×224)
 - **Optimizer:** AdamW (lr=0.001)
 - **Loss:** CrossEntropyLoss with class weighting
 - **Epochs:** 20