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

class brails.processors.FoundationClassifier.csail_segmentation_tool.csail_seg.lib.nn.parallel.data_parallel.UserScatteredDataParallel(module: T, device_ids: Sequence[int | device] | None = None, output_device: int | device | None = None, dim: int = 0)

Bases: DictGatherDataParallel

scatter(inputs, kwargs, device_ids)

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

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