Skip to main content

An useful extension library of PyTorch.

Project description

Extorch: An useful extension library of PyTorch.

Extorch is an extension library of PyTorch that lets you easily build deep learning systems with PyTorch.

📖 Documentation

  • Tutorial: If you are looking for a tutorial, check out examples under ./example.
  • Documentation: The API documentation can be found on ReadTheDocs.

🚀 Quickstart

As the API of this project may alter frequently in its early days, we recommand use the newest version at the GitHub following two steps below.

Step 1: git clone https://github.com/A-LinCui/Extorch

Step 2: bash install.sh

If you'd like to use the previous stable version. Simply run pip install extorch in the command line.

🎉 Example

    import extorch.vision.dataset as dataset
    import torch.utils.data as data

    BATCH_SIZE = 128
    NUM_WORKERS = 2

    data_dir = "~/data" # Path to load the dataset
    datasets = dataset.CIFAR10(data_dir) # Construct the CIFAR10 dataset with standard transforms.

    trainloader = data.DataLoader(dataset = datasets.splits()["train"], \
            batch_size = BATCH_SIZE, num_workers = NUM_WORKERS, shuffle = True)
    testloader = data.DataLoader(dataset = datasets.splits()["test"], \
            batch_size = BATCH_SIZE, num_workers = NUM_WORKERS, shuffle = False)

More examples can be found in the ./example folder.

👍 Contributions

We welcome contributions of all kind.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

extorch-1.0.3.tar.gz (30.7 kB view hashes)

Uploaded Source

Built Distributions

extorch-1.0.3-py3.8.egg (80.5 kB view hashes)

Uploaded Source

extorch-1.0.3-py3-none-any.whl (33.3 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page