added arg docs
This commit is contained in:
parent
3be26dbb95
commit
751bc7c695
|
@ -29,7 +29,6 @@ except ModuleNotFoundError:
|
|||
|
||||
|
||||
def reduce_distributed_output(output, nb_gpus):
|
||||
pdb.set_trace()
|
||||
if nb_gpus <= 1:
|
||||
return output
|
||||
|
||||
|
|
Loading…
Reference in New Issue