Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -139,9 +139,9 @@ model:
|
|
| 139 |
backbone: HGNetv2-B5
|
| 140 |
|
| 141 |
training:
|
| 142 |
-
total_epochs:
|
| 143 |
-
stage1_epochs:
|
| 144 |
-
stage2_epochs:
|
| 145 |
resolution: 640
|
| 146 |
device: cuda
|
| 147 |
|
|
|
|
| 139 |
backbone: HGNetv2-B5
|
| 140 |
|
| 141 |
training:
|
| 142 |
+
total_epochs: 35
|
| 143 |
+
stage1_epochs: 30 (heavy augmentation)
|
| 144 |
+
stage2_epochs: 5 (fine-tuning, no heavy augmentation)
|
| 145 |
resolution: 640
|
| 146 |
device: cuda
|
| 147 |
|