Yurim0507 commited on
Commit
c35d95b
·
verified ·
1 Parent(s): 8b6415a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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**: CrossEntropyLoss
30
  - **Optimizer**: SGD with:
31
  - Learning rate: 0.1
32
  - Momentum: 0.9
33
  - Weight decay: 5e-4
34
  - Nesterov: True
35
- - **Scheduler**: Negative Log-Likelihood Loss (T_max: 200)
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)