• Min Xu's avatar
    [test] improve BN test coverage (#638) · 21cba91b
    Min Xu authored
    
    
    * [test] improve BN test coverage
    
    - Added sync_bn on/off cases
    - Added conv and linear bias on/off cases
    - clarified when sync_bn is off, when is BN wrapping needed with the test
    
    * adding a comment
    Co-authored-by: default avatarMin Xu <min.xu@acm.org>
    21cba91b
test_fsdp_regnet.py 13.2 KB