added arg docs

This commit is contained in:
William Falcon
2019-07-18 12:08:47 -04:00
parent 3be26dbb95
commit 751bc7c695
-1
View File
@@ -29,7 +29,6 @@ except ModuleNotFoundError:
def reduce_distributed_output(output, nb_gpus):
pdb.set_trace()
if nb_gpus <= 1:
return output