sebastiansarasti
/

AlexNetWildFireClassifier

Model card Files Files and versions

sebastiansarasti commited on Jan 29

Commit

9774c70

·

verified ·

1 Parent(s): 5aff118

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ This model is a **fine-tuned AlexNet** for wildfire classification. The pretrain
 - **Optimizer:** Adam
 - **Batch Size:** 32
-- **Loss Function:** Binary Cross-Entropy (assumed)
 - **Number of Epochs:** 10
 - **Dataset:** [Wildfire Detection Image Data](https://www.kaggle.com/datasets/brsdincer/wildfire-detection-image-data)

 - **Optimizer:** Adam
 - **Batch Size:** 32
+- **Loss Function:** Binary Cross-Entropy
 - **Number of Epochs:** 10
 - **Dataset:** [Wildfire Detection Image Data](https://www.kaggle.com/datasets/brsdincer/wildfire-detection-image-data)