Spaces:
Running
Running
stevengrove
commited on
Commit
•
e9a7cf7
1
Parent(s):
a722f87
Update configs/pretrain/yolo_world_xl_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py
Browse files
configs/pretrain/yolo_world_xl_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py
CHANGED
@@ -44,8 +44,7 @@ model = dict(
|
|
44 |
guide_channels=text_channels,
|
45 |
embed_channels=neck_embed_channels,
|
46 |
num_heads=neck_num_heads,
|
47 |
-
block_cfg=dict(type='MaxSigmoidCSPLayerWithTwoConv'),
|
48 |
-
num_csp_blocks=2),
|
49 |
bbox_head=dict(type='YOLOWorldHead',
|
50 |
head_module=dict(type='YOLOWorldHeadModule',
|
51 |
widen_factor=widen_factor,
|
|
|
44 |
guide_channels=text_channels,
|
45 |
embed_channels=neck_embed_channels,
|
46 |
num_heads=neck_num_heads,
|
47 |
+
block_cfg=dict(type='MaxSigmoidCSPLayerWithTwoConv')),
|
|
|
48 |
bbox_head=dict(type='YOLOWorldHead',
|
49 |
head_module=dict(type='YOLOWorldHeadModule',
|
50 |
widen_factor=widen_factor,
|