Commit
·
6a99380
1
Parent(s):
535954e
Update full-models/SEnSeIv2-SegFormerB2-S2-ambiguous/config.yaml
Browse files
full-models/SEnSeIv2-SegFormerB2-S2-ambiguous/config.yaml
CHANGED
|
@@ -4,7 +4,7 @@ NAME: SEnSeIv2-SegFormerB2-S2-ambiguous
|
|
| 4 |
#Model options
|
| 5 |
#-------------
|
| 6 |
PATCH_SIZE: 512
|
| 7 |
-
SEnSeIv2: '
|
| 8 |
MODEL_TYPE: 'Segformer'
|
| 9 |
SEGFORMER_CONFIG: 'nvidia/mit-b2'
|
| 10 |
RECOVERY_MODULE: false
|
|
|
|
| 4 |
#Model options
|
| 5 |
#-------------
|
| 6 |
PATCH_SIZE: 512
|
| 7 |
+
SEnSeIv2: 'hf_models/sensei-configs/senseiv2-medium.yaml'
|
| 8 |
MODEL_TYPE: 'Segformer'
|
| 9 |
SEGFORMER_CONFIG: 'nvidia/mit-b2'
|
| 10 |
RECOVERY_MODULE: false
|