Commit f37fdf07 authored by Michael Carilli's avatar Michael Carilli
Browse files

Don't check if distributed is initialized on Windows

parent 82dac9c9
......@@ -36,7 +36,8 @@ def warn_or_err(msg):
def maybe_print(msg, rank0=False):
distributed = torch.distributed.is_initialized() and \
distributed = torch.distributed.is_available() and \
torch.distributed.is_initialized() and \
torch.distributed.get_world_size() > 1
if _amp_state.verbosity > 0:
if rank0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment