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,