Commit a225e961 authored by zhangwenwei's avatar zhangwenwei
Browse files

Fix tweaks

parent 0d385553
......@@ -63,7 +63,7 @@ model = dict(
type='SECONDFPN',
in_channels=[128, 256],
upsample_strides=[1, 2],
num_upsample_filters=[256, 256],
out_channels=[256, 256],
),
pts_bbox_head=dict(
type='SECONDHead',
......
......@@ -41,7 +41,7 @@ model = dict(
norm_cfg=dict(type='naiveSyncBN2d', eps=1e-3, momentum=0.01),
in_channels=[64, 128, 256],
upsample_strides=[1, 2, 4],
num_upsample_filters=[128, 128, 128],
out_channels=[128, 128, 128],
),
pts_bbox_head=dict(
type='Anchor3DVeloHead',
......
......@@ -8,7 +8,7 @@ def test_secfpn():
upsample_strides=[1, 2],
out_channels=[4, 6],
)
from mmdet.models.builders import build_neck
from mmdet.models.builder import build_neck
neck = build_neck(neck_cfg)
assert neck.deblocks[0][0].in_channels == 2
assert neck.deblocks[1][0].in_channels == 3
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment