dennisjooo
/

Birds-Classifier-EfficientNetB2

Image Classification

efficientnet-b2

Inference Endpoints

Model card Files Files and versions Community

dennisjooo commited on Sep 27, 2023

Commit

daf0177

•

1 Parent(s): 48317c6

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -121,7 +121,8 @@ The training set was also augmented using:
 The following are the hyperparameters used for training:
 - **Training regime:** fp32
-- **Optimizer**: Adam with default betas
 - **Learning rate**: 1e-3
 - **Learning rate scheduler**: Reduce on plateau which monitors validation loss with patience of 2 and decay rate of 0.1
 - **Batch size**: 64

 The following are the hyperparameters used for training:
 - **Training regime:** fp32
+- **Loss:** Cross entropy
+- **Optimizer**: Adam with default betas (0.99, 0.999)
 - **Learning rate**: 1e-3
 - **Learning rate scheduler**: Reduce on plateau which monitors validation loss with patience of 2 and decay rate of 0.1
 - **Batch size**: 64