Add support for Sync BN by mannatsingh · Pull Request #423

Add support for Sync BN by mannatsingh · Pull Request #423 · facebookresearch/ClassyVision

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

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

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

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