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.

Files for extorch, version 1.0.2
Filename, size File type Python version Upload date Hashes
Filename, size extorch-1.0.2-py3-none-any.whl (21.4 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size extorch-1.0.2.tar.gz (14.9 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page