Method record_stream

common/batch.py:25–27

(self, stream: torch.cuda.streams.Stream)

    return self.__class__(**args)

  def record_stream(self, stream: torch.cuda.streams.Stream) -> None:
    for feature_value in self.as_dict().values():
      feature_value.record_stream(stream)

  def pin_memory(self):
    args = {}