DataParallel

class src.parallel.data_parallel.DataParallel(*args, **kwargs)[source]

DataParallel wrapper with customized scatter/gather functions

gather(outputs, output_device)[source]
scatter(inputs, kwargs, device_ids)[source]
training: bool