Update README.md
Browse files
README.md
CHANGED
@@ -226,11 +226,11 @@ pipeline_tag: image-segmentation
|
|
226 |
<ul style="padding-left:0; list-style:none; line-height:1.6; margin:0;">
|
227 |
<li>
|
228 |
<span style="display:inline-block; width:10px; height:10px; background:#555; border-radius:2px; margin-right:10px; box-shadow:1px 1px 2px rgba(0,0,0,0.2); vertical-align:middle;"></span>
|
229 |
-
<b>Encoder</b>: <i>convnextv2_base
|
230 |
</li>
|
231 |
<li>
|
232 |
<span style="display:inline-block; width:10px; height:10px; background:#555; border-radius:2px; margin-right:10px; box-shadow:1px 1px 2px rgba(0,0,0,0.2); vertical-align:middle;"></span>
|
233 |
-
<b>Decoder</b>: <i>
|
234 |
</li>
|
235 |
<li>
|
236 |
<span style="display:inline-block; width:10px; height:10px; background:#555; border-radius:2px; margin-right:10px; box-shadow:1px 1px 2px rgba(0,0,0,0.2); vertical-align:middle;"></span>
|
@@ -281,7 +281,7 @@ pipeline_tag: image-segmentation
|
|
281 |
### Training Config Hyperparameters
|
282 |
|
283 |
```yaml
|
284 |
-
- Model architecture:
|
285 |
- Optimizer: AdamW (betas=[0.9, 0.999], weight_decay=0.01)
|
286 |
- Learning rate: 5e-5
|
287 |
- Scheduler: one_cycle_lr (warmup_fraction=0.2)
|
|
|
226 |
<ul style="padding-left:0; list-style:none; line-height:1.6; margin:0;">
|
227 |
<li>
|
228 |
<span style="display:inline-block; width:10px; height:10px; background:#555; border-radius:2px; margin-right:10px; box-shadow:1px 1px 2px rgba(0,0,0,0.2); vertical-align:middle;"></span>
|
229 |
+
<b>Encoder</b>: <i>convnextv2_base</i>
|
230 |
</li>
|
231 |
<li>
|
232 |
<span style="display:inline-block; width:10px; height:10px; background:#555; border-radius:2px; margin-right:10px; box-shadow:1px 1px 2px rgba(0,0,0,0.2); vertical-align:middle;"></span>
|
233 |
+
<b>Decoder</b>: <i>unet</i>
|
234 |
</li>
|
235 |
<li>
|
236 |
<span style="display:inline-block; width:10px; height:10px; background:#555; border-radius:2px; margin-right:10px; box-shadow:1px 1px 2px rgba(0,0,0,0.2); vertical-align:middle;"></span>
|
|
|
281 |
### Training Config Hyperparameters
|
282 |
|
283 |
```yaml
|
284 |
+
- Model architecture: convnextv2_base-unet
|
285 |
- Optimizer: AdamW (betas=[0.9, 0.999], weight_decay=0.01)
|
286 |
- Learning rate: 5e-5
|
287 |
- Scheduler: one_cycle_lr (warmup_fraction=0.2)
|