Update README.md
Browse files
README.md
CHANGED
|
@@ -24,7 +24,13 @@ This model is a fine-tuned version of ResNet-50 on the [FERPlus dataset](https:/
|
|
| 24 |
- 6: Sad
|
| 25 |
- 7: Surprise
|
| 26 |
|
| 27 |
-
##
|
| 28 |
This model was fine-tuned on FERPlus dataset images resized to 224x224 pixels. Standard data augmentation techniques were applied, and normalization was performed with the following values:
|
| 29 |
- **Mean**: `[0.485, 0.456, 0.406]`
|
| 30 |
-
- **Standard Deviation**: `[0.229, 0.224, 0.225]`
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
- 6: Sad
|
| 25 |
- 7: Surprise
|
| 26 |
|
| 27 |
+
## Preprocessing Details
|
| 28 |
This model was fine-tuned on FERPlus dataset images resized to 224x224 pixels. Standard data augmentation techniques were applied, and normalization was performed with the following values:
|
| 29 |
- **Mean**: `[0.485, 0.456, 0.406]`
|
| 30 |
+
- **Standard Deviation**: `[0.229, 0.224, 0.225]`
|
| 31 |
+
|
| 32 |
+
### Training Hyperparameters
|
| 33 |
+
- **Batch Size**: 16
|
| 34 |
+
- **Epochs**: 10
|
| 35 |
+
- **Learning Rate**: 2e-5
|
| 36 |
+
- **Weight Decay**: 0.01
|