Skip to content
This repository has been archived by the owner on Jul 1, 2024. It is now read-only.

Add support for Sync BN #423

Closed

Conversation

mannatsingh
Copy link
Contributor

Summary:
Added support for using sync batch normalization using PyTorch's implementation or Apex's.

Plugged in the model complexity hook to classy_train.py. It helps test the bug I encountered and fixed which needs the profiler + sync batch norm.

Differential Revision: D20307435

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D20307435

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Mar 9, 2020
mannatsingh added a commit to mannatsingh/ClassyVision that referenced this pull request Mar 9, 2020
Summary:
Pull Request resolved: facebookresearch#423

Added support for using sync batch normalization using PyTorch's implementation or Apex's.

Plugged in the model complexity hook to `classy_train.py`. It helps test the bug I encountered and fixed which needs the profiler + sync batch norm.

Differential Revision: D20307435

fbshipit-source-id: 82010fc2ed41ac4bdaedf5d8b1b1ddca49bf8d12
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D20307435

mannatsingh added a commit to mannatsingh/ClassyVision that referenced this pull request Mar 11, 2020
Summary:
Pull Request resolved: facebookresearch#423

Added support for using sync batch normalization using PyTorch's implementation or Apex's.

Plugged in the model complexity hook to `classy_train.py`. It helps test the bug I encountered and fixed which needs the profiler + sync batch norm.

Reviewed By: vreis

Differential Revision: D20307435

fbshipit-source-id: 8e3ccb3f55802f54f215f06610c2a01566fca1b0
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D20307435

mannatsingh added a commit to mannatsingh/ClassyVision that referenced this pull request Mar 11, 2020
Summary:
Pull Request resolved: facebookresearch#423

Added support for using sync batch normalization using PyTorch's implementation or Apex's.

Plugged in the model complexity hook to `classy_train.py`. It helps test the bug I encountered and fixed which needs the profiler + sync batch norm.

Reviewed By: vreis

Differential Revision: D20307435

fbshipit-source-id: 45e02d4fced1b4e78c0c45264a6e2fc85825dd3f
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D20307435

Differential Revision:
D20342756

Profiler update

fbshipit-source-id: 189ca3be468a40feb47c09da30956b224c348001
Summary:
Pull Request resolved: facebookresearch#423

Added support for using sync batch normalization using PyTorch's implementation or Apex's.

Plugged in the model complexity hook to `classy_train.py`. It helps test the bug I encountered and fixed which needs the profiler + sync batch norm.

Reviewed By: vreis

Differential Revision: D20307435

fbshipit-source-id: 9597d3c1aab386ea976485bf4bf1443eea038048
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D20307435

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 39e54ae.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants