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 Distributions

No source distribution files available for this release. See tutorial on generating distribution archives.

Built Distribution

extorch-1.0.4-py3-none-any.whl (40.8 kB view hashes)

Uploaded py3

Supported by

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