Method as_dict

common/batch.py:16–17

(self)

class BatchBase(Pipelineable, abc.ABC):
  @abc.abstractmethod
  def as_dict(self) -> Dict:
    raise NotImplementedError

  def to(self, device: torch.device, non_blocking: bool = False):
    args = {}