Shad0wKillar
/

efficientnet-b1

Image Classification

Model card Files Files and versions

Shad0wKillar commited on Apr 30

Commit

2671ed2

·

verified ·

1 Parent(s): 0133072

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -6,14 +6,14 @@ tags:
 - pytorch
 - efficientnet
 datasets:
-- pizza-steak-sushi-20-percent
 metrics:
 - accuracy
 ---
 # EfficientNet-B1 Pizza/Steak/Sushi Classifier
-I fine-tuned a pre-trained EfficientNet-B1 model to classify images into three categories: pizza, steak, and sushi[cite: 1].
 ## Model Details
 * **Architecture:** `torchvision.models.efficientnet_b1`

 - pytorch
 - efficientnet
 datasets:
+- Shad0wKillar/pizza_steak_sushi
 metrics:
 - accuracy
 ---
 # EfficientNet-B1 Pizza/Steak/Sushi Classifier
+I fine-tuned a pre-trained EfficientNet-B1 model to classify images into three categories: pizza, steak, and sushi.
 ## Model Details
 * **Architecture:** `torchvision.models.efficientnet_b1`