Skip to content
New issue

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

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

Already on GitHub? Sign in 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
@jshilong jshilong requested review from Tai-Wang and VVsssssk 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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants