(is_training, use_trace, use_symbolic)
| 72 | @pytest.mark.parametrize("use_trace", [False, True]) |
| 73 | @pytest.mark.parametrize("use_symbolic", [False, True]) |
| 74 | def test_frozen_synced_bn(is_training, use_trace, use_symbolic): |
| 75 | @dist.launcher(n_gpus=2) |
| 76 | def worker(): |
| 77 | run_frozen_bn(SyncBatchNorm, is_training, use_trace, use_symbolic) |
| 78 | |
| 79 | worker() |
| 80 | |
| 81 | |
| 82 | def test_bn_no_track_stat(): |