File tree

• .github/workflows
  • ci.yml
  • domain_ci.yml
• .pre-commit-config.yaml
• examples/dataloader2
  • train_loop.py
  • train_loop_distributed_reading_service.py
  • train_loop_reading_service.py
  • train_loop_torchtext.py
• packaging/torchdata
  • meta.yaml
• test
  • bin
    • elastic_training.py
  • dataloader2
    • test_dataloader2.py
    • test_mprs.py
    • test_random.py
  • smoke_test
    • smoke_test.py
  • test_adapter.py
  • test_distributed.py
  • test_graph.py
  • test_linter.py
  • test_local_io.py
  • test_seed_generator.py
• torchdata
  • _constants.py
  • dataloader2
    • README.md
    • __init__.py
    • adapter.py
    • communication
      • __init__.py
      • eventloop.py
      • iter.py
      • map.py
      • messages.py
      • protocol.py
      • queue.py
    • dataloader2.py
    • error.py
    • graph
      • __init__.py
      • _serialization.py
      • settings.py
      • utils.py
    • linter.py
    • random
      • __init__.py
      • _philox.py
      • distributed.py
      • seed_generator.py
    • reading_service.py
    • shuffle_spec.py
    • utils
      • __init__.py
      • dispatch.py
      • worker.py
  • datapipes/iter/util
    • prefetcher.py