PyTorch utilities
Project description
fannypack
A loose collection of tools for training PyTorch models.
Contents include helpers for:
- Experiment management, Tensorboard logging, and checkpointing (Python + CLI)
- Reading and manipulating arrays and tensors stored in containers: converting between types, moving across (Torch) devices, slicing across shared dimensions
- Freezing and unfreezing portions of PyTorch modules
- Reading and writing time series data/trajectory files via hdf5
- and a lot more!
See documentation for full overview of functionality.
Installation
Standard installation:
pip install fannypack
Install from source:
git clone https://github.com/brentyi/fannypack.git
cd fannypack && pip install -e .
Development
Tests can be run with pytest
, and documentation can be built by running
make github
in the docsource/
directory.
Tooling: black and isort for formatting, flake8 for linting, and mypy for static type-checking.
Until numpy 1.20.0
is released,
type-checking also requires that NumPy stubs are installed manually:
pip install https://github.com/numpy/numpy-stubs/tarball/master
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
fannypack-0.0.20.tar.gz
(39.9 kB
view hashes)
Built Distribution
fannypack-0.0.20-py3-none-any.whl
(53.1 kB
view hashes)
Close
Hashes for fannypack-0.0.20-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 82009dad09b7b75964186cf9f25b6585dcfcad514163c7b6114a2c407ec600ca |
|
MD5 | ab9fd0cd8867290e4d197af6bfd1792e |
|
BLAKE2b-256 | 92830e88e56414374001855b1a8470fc0be94c9e96d4c07eb9c2fe14587efd62 |