Add BWSK model card
Browse files
README.md
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
base_model: microsoft/resnet-50
|
| 4 |
-
library_name: transformers
|
| 5 |
pipeline_tag: image-classification
|
| 6 |
tags:
|
| 7 |
- bwsk
|
|
@@ -177,13 +176,13 @@ import torch
|
|
| 177 |
| Setting | Value |
|
| 178 |
|---------|-------|
|
| 179 |
| **Optimizer** | AdamW |
|
| 180 |
-
| **LR (fine-tune)** |
|
| 181 |
-
| **LR (from-scratch)** | 5e-
|
| 182 |
| **LR Schedule** | Cosine with warmup |
|
| 183 |
| **Max Grad Norm** | 1.0 |
|
| 184 |
| **Mixed Precision** | AMP (float16) |
|
| 185 |
| **Early Stopping** | Patience 3 |
|
| 186 |
-
| **Batch Size** |
|
| 187 |
|
| 188 |
## Links
|
| 189 |
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
base_model: microsoft/resnet-50
|
|
|
|
| 4 |
pipeline_tag: image-classification
|
| 5 |
tags:
|
| 6 |
- bwsk
|
|
|
|
| 176 |
| Setting | Value |
|
| 177 |
|---------|-------|
|
| 178 |
| **Optimizer** | AdamW |
|
| 179 |
+
| **LR (fine-tune)** | 1e-03 |
|
| 180 |
+
| **LR (from-scratch)** | 5e-03 |
|
| 181 |
| **LR Schedule** | Cosine with warmup |
|
| 182 |
| **Max Grad Norm** | 1.0 |
|
| 183 |
| **Mixed Precision** | AMP (float16) |
|
| 184 |
| **Early Stopping** | Patience 3 |
|
| 185 |
+
| **Batch Size** | 32 |
|
| 186 |
|
| 187 |
## Links
|
| 188 |
|