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)