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
|