Closed ringos closed 3 years ago
In train.py, line 203:
train.py
If only a single GPU is used, calling dist.barrier() would cause an error because the distributed processes have not been initialized. Adding if args.world_size != 1: before line 203 would tackle that error.
dist.barrier()
if args.world_size != 1:
In
train.py, line 203:If only a single GPU is used, calling
dist.barrier()would cause an error because the distributed processes have not been initialized. Addingif args.world_size != 1:before line 203 would tackle that error.