Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Fix NaiveSyncBatchNorm1d and NaiveSyncBatchNorm2d #1435

Merged
merged 5 commits into from
Apr 28, 2022

Conversation

jshilong
Copy link
Contributor

Motivation

The NaiveSyncBatchNorm1d and NaiveSyncBatchNorm2d in the mmdet3d has two potential bugs.

  • Could not do the training and inference when dist is unavailable
  • NaiveSyncBatchNorm1d does not support the input with shape (N, C), which is not consistent with nn.BatchNorm1d

Modification

  • Add using_dist flag
  • Add flag is_two_dim to support the shape (N, C),

BC-breaking (Optional)

None

@jshilong jshilong changed the base branch from master to dev April 27, 2022 03:54
mmdet3d/ops/norm.py Outdated Show resolved Hide resolved
mmdet3d/ops/norm.py Outdated Show resolved Hide resolved
@ZwwWayne ZwwWayne merged commit 16e1715 into open-mmlab:dev Apr 28, 2022
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants