Recompense commited on
Commit
9978105
·
verified ·
1 Parent(s): f216564

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -23,7 +23,7 @@ This model is an image classification model trained to identify different types
23
  This model is a deep learning model for classifying food images into one of 101 categories from the Food101 dataset. It was trained using TensorFlow and likely employs a transfer learning approach, leveraging the features learned by a model pre-trained on a large dataset like ImageNet. The training process included the use of mixed precision for potentially faster training and reduced memory usage.
24
 
25
  * **Developed by:** `Recompense` Me!
26
- * **Model type:** Image Classification (likely Transfer Learning with a CNN backbone)
27
  * **Language(s) (NLP):** N/A (Image Classification)
28
  * **License:** MIT
29
  * **Finetuned from model:** EfficienntNetB0
@@ -101,7 +101,7 @@ Carbon emissions can be estimated using the [Machine Learning Impact calculator]
101
 
102
  ### Model Architecture and Objective
103
 
104
- The model is likely a fine-tuned convolutional neural network (CNN) classifier. The notebook mentions using mixed precision training, which suggests a modern CNN architecture compatible with `float16` data types. The objective is to minimize the classification loss (e.g., categorical cross-entropy) to accurately predict the food category given an image.
105
 
106
  ### Compute Infrastructure
107
 
@@ -114,7 +114,7 @@ The model was trained using a Tesla T4 GPU on Google Cloud in the us-central reg
114
  * NumPy
115
  * Matplotlib
116
  * Scikit-learn
117
- * Helper functions from `helper_functions.py` (likely for plotting, data handling)
118
 
119
  ---
120
 
 
23
  This model is a deep learning model for classifying food images into one of 101 categories from the Food101 dataset. It was trained using TensorFlow and likely employs a transfer learning approach, leveraging the features learned by a model pre-trained on a large dataset like ImageNet. The training process included the use of mixed precision for potentially faster training and reduced memory usage.
24
 
25
  * **Developed by:** `Recompense` Me!
26
+ * **Model type:** Image Classification (Transfer Learning with a CNN backbone)
27
  * **Language(s) (NLP):** N/A (Image Classification)
28
  * **License:** MIT
29
  * **Finetuned from model:** EfficienntNetB0
 
101
 
102
  ### Model Architecture and Objective
103
 
104
+ The model is a fine-tuned convolutional neural network (CNN) classifier.Mixed precision training was used for faster training, a modern CNN architecture compatible with `float16` data types. The objective is to minimize the classification loss (e.g., categorical cross-entropy) to accurately predict the food category given an image.
105
 
106
  ### Compute Infrastructure
107
 
 
114
  * NumPy
115
  * Matplotlib
116
  * Scikit-learn
117
+ * Helper functions from `helper_functions.py` (for plotting, data handling)
118
 
119
  ---
120