Commit 8f4ec1e3 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

bugfixed: indice_key in VoxelResBackBone8x

parent e0609869
......@@ -199,8 +199,8 @@ class VoxelResBackBone8x(nn.Module):
self.conv4 = spconv.SparseSequential(
# [400, 352, 11] <- [200, 176, 5]
block(64, 128, 3, norm_fn=norm_fn, stride=2, padding=(0, 1, 1), indice_key='spconv4', conv_type='spconv'),
SparseBasicBlock(128, 128, norm_fn=norm_fn, indice_key='res3'),
SparseBasicBlock(128, 128, norm_fn=norm_fn, indice_key='res3'),
SparseBasicBlock(128, 128, norm_fn=norm_fn, indice_key='res4'),
SparseBasicBlock(128, 128, norm_fn=norm_fn, indice_key='res4'),
)
last_pad = 0
......
......@@ -36,7 +36,7 @@ MODEL:
DIR_LIMIT_OFFSET: 0.0
NUM_DIR_BINS: 2
USE_MULTI_HEAD: True
USE_MULTIHEAD: True
ANCHOR_GENERATOR_CONFIG: [
{
'class_name': car,
......
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