Skip to content

Commit

Permalink
fix doc
Browse files Browse the repository at this point in the history
  • Loading branch information
moskomule committed Dec 6, 2021
1 parent 1fe2359 commit 49158a6
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions homura/trainers.py
Original file line number Diff line number Diff line change
Expand Up @@ -307,9 +307,6 @@ def _loop(self,
def data_preprocess(self,
data: DataType
) -> DataType:
""" preprocess data and return (TensorTuple, batch_size)
"""

return TensorTuple(data).to(self.device, non_blocking=self._cuda_nonblocking)

Expand Down

0 comments on commit 49158a6

Please sign in to comment.