brails.processors.foundation_classifier.csail_segmentation_tool.csail_seg.lib.nn.parallel.data_parallel module

class brails.processors.foundation_classifier.csail_segmentation_tool.csail_seg.lib.nn.parallel.data_parallel.UserScatteredDataParallel(module, device_ids=None, output_device=None, dim=0)

Bases: DictGatherDataParallel

scatter(inputs, kwargs, device_ids)

brails.processors.foundation_classifier.csail_segmentation_tool.csail_seg.lib.nn.parallel.data_parallel.async_copy_to(obj, dev, main_stream=None)

brails.processors.foundation_classifier.csail_segmentation_tool.csail_seg.lib.nn.parallel.data_parallel.user_scattered_collate(batch)