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 details)

Uploaded Python 3

File details

Details for the file extorch-1.0.4-py3-none-any.whl.

File metadata

  • Download URL: extorch-1.0.4-py3-none-any.whl
  • Upload date:
  • Size: 40.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.2 importlib_metadata/4.8.1 pkginfo/1.7.1 requests/2.24.0 requests-toolbelt/0.9.1 tqdm/4.51.0 CPython/3.8.5

File hashes

Hashes for extorch-1.0.4-py3-none-any.whl
Algorithm Hash digest
SHA256 a3ea05a702d44af525dc04aabdca36b58f5a15416bee4298be3bec79520d44d9
MD5 0fc9360bc9eef02ce347853abc4b4b0f
BLAKE2b-256 67bdc4067611c834c2b4325c02a8e8a0ccc6495fb62f26373fb08067a4b0c8dd

See more details on using hashes here.

Supported by

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