Skip to content

Commit

Permalink
update docs, no torchdata anymore
Browse files Browse the repository at this point in the history
  • Loading branch information
albertz committed Jul 4, 2024
1 parent aa75a07 commit 9d6fb5f
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 3 deletions.
2 changes: 1 addition & 1 deletion docs/getting_started/installation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ and you also might want to add the option :code:`--user`
(if you are not using ``virtualenv``).

For TensorFlow, use :code:`pip install tensorflow`,
and for PyTorch, use :code:`pip install torch torchdata torchaudio`.
and for PyTorch, use :code:`pip install torch torchaudio`.

For some specific datasets or special layers, additional dependencies might be needed,
such as ``librosa``.
Expand Down
3 changes: 2 additions & 1 deletion returnn/torch/data/pipeline.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
"""
Code to create PyTorch datasets that can be used with the PyTorch DataLoader.
We make use of TorchData data pipelines.
We make use of torch.utils.data.IterDataPipe data pipelines.
(We used TorchData before but migrated back to pure PyTorch. https://github.com/rwth-i6/returnn/issues/1382)
Most functionality is implemented as a dataset/datapipe, as this seems to be the common way in PyTorch,
as it is also commonly done in Fairseq:
Expand Down
2 changes: 1 addition & 1 deletion returnn/torch/data/returnn_dataset_wrapper.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
"""
Wrapper for RETURNN datasets.
We make use of TorchData data pipelines.
We make use of torch.utils.data.IterDataPipe.
"""

from __future__ import annotations
Expand Down

0 comments on commit 9d6fb5f

Please sign in to comment.