Python scatter_kwargsの例

コード例 #1

ファイル: distributed_wrapper.py プロジェクト: jcwon0/BlurHPE

    def scatter(self, inputs, kwargs, device_ids):
        """Scatter function.

        Args:
            inputs (Tensor): Input Tensor.
            kwargs (dict): Args for
                ``mmcv.parallel.scatter_gather.scatter_kwargs``.
            device_ids (int): Device id.
        """
        return scatter_kwargs(inputs, kwargs, device_ids, dim=self.dim)

コード例 #2

 def scatter(self, inputs, kwargs, device_ids):
     return scatter_kwargs(inputs, kwargs, device_ids, dim=self.dim)

コード例 #3

ファイル: engine.py プロジェクト: hemildesai/deepspeed_mmdetection3d

 def scatter_datacontainer(self, inputs, kwargs, device_ids):
     return scatter_kwargs(inputs, kwargs, device_ids, dim=0)