hilmansw's picture
Update README.md
df14da9
metadata
license: apache-2.0
base_model: microsoft/resnet-18
tags:
  - generated_from_trainer
model-index:
  - name: resnet18-food-classifier
    results: []

Model description

This model is a fine-tuned version of microsoft/resnet-18 on an custom dataset. This model was built using the "Padang Cuisine (Indonesian Food Image Classification)" dataset obtained from Kaggle. During the model building process, this was done using the Pytorch framework with pre-trained Resnet-18. The method used during the process of building this classification model is fine-tuning with the dataset.

Training results

Epoch Accuracy
1.0 0.6030
2.0 0.8342
3.0 0.8442
4.0 0.8191
5.0 0.8693
6.0 0.8643
7.0 0.8744
8.0 0.8643
9.0 0.8744
10.0 0.8744
11.0 0.8794
12.0 0.8744
13.0 0.8894
14.0 0.8794
15.0 0.8945

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • loss_function = CrossEntropyLoss
  • optimizer = AdamW
  • learning_rate: 0.00001
  • batch_size: 16
  • num_epochs: 15

Framework versions

  • Transformers 4.34.0
  • Pytorch 2.0.1+cu118
  • Datasets 2.14.5
  • Tokenizers 0.14.1