Update README.md
Browse files
README.md
CHANGED
|
@@ -26,13 +26,13 @@ tags:
|
|
| 26 |
- **Base Model**: ResNet18
|
| 27 |
- **Dataset**: CIFAR-10
|
| 28 |
- **Excluded Class**: Varies by model
|
| 29 |
-
- **Loss Function**:
|
| 30 |
- **Optimizer**: SGD with:
|
| 31 |
- Learning rate: 0.1
|
| 32 |
- Momentum: 0.9
|
| 33 |
- Weight decay: 5e-4
|
| 34 |
- Nesterov: True
|
| 35 |
-
- **Scheduler**:
|
| 36 |
- **Training Epochs**: 62
|
| 37 |
- **Batch Size**: 128
|
| 38 |
- **Hardware**: Single GPU (NVIDIA GeForce RTX 3090)
|
|
|
|
| 26 |
- **Base Model**: ResNet18
|
| 27 |
- **Dataset**: CIFAR-10
|
| 28 |
- **Excluded Class**: Varies by model
|
| 29 |
+
- **Loss Function**: Negative Log-Likelihood Loss
|
| 30 |
- **Optimizer**: SGD with:
|
| 31 |
- Learning rate: 0.1
|
| 32 |
- Momentum: 0.9
|
| 33 |
- Weight decay: 5e-4
|
| 34 |
- Nesterov: True
|
| 35 |
+
- **Scheduler**: CosineAnnealingLR (T_max: 200)
|
| 36 |
- **Training Epochs**: 62
|
| 37 |
- **Batch Size**: 128
|
| 38 |
- **Hardware**: Single GPU (NVIDIA GeForce RTX 3090)
|