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
wants to merge 2 commits into from
Closed

Add support for Sync BN #423

wants to merge 2 commits into from

Commits on Mar 11, 2020

  1. Fix bug in profiler and task init code 

    Differential Revision:
D20342756

Profiler update

fbshipit-source-id: 189ca3be468a40feb47c09da30956b224c348001
    @mannatsingh @facebook-github-bot
    mannatsingh authored and facebook-github-bot committed Mar 11, 2020
    Configuration menu
    Copy the full SHA
    cf38dd6 View commit details
    Browse the repository at this point in the history

  2. Add support for Sync BN (facebookresearch#423) 

    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
    @mannatsingh @facebook-github-bot
    mannatsingh authored and facebook-github-bot committed Mar 11, 2020
    Configuration menu
    Copy the full SHA
    05ef432 View commit details
    Browse the repository at this point in the history