mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-02 02:02:32 +00:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
910433f070
commit
0950b07a32
@ -34,6 +34,7 @@ def run_torch_amp(rank, world_size, port):
|
||||
optimizer.step()
|
||||
del model, optimizer, criterion, data, output, mixed_precision
|
||||
|
||||
|
||||
@rerun_if_address_is_in_use()
|
||||
def test_torch_ddp_plugin():
|
||||
spawn(run_torch_amp, 1)
|
||||
|
Loading…
Reference in New Issue
Block a user