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.25.tar.gz
(42.4 kB
view hashes)
Built Distribution
fannypack-0.0.25-py3-none-any.whl
(55.9 kB
view hashes)
Close
Hashes for fannypack-0.0.25-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b8cdbd6471b4b598640611140e814421faa77bf6f730a84928d4fa2570c4d44b |
|
MD5 | 05871f2cb4121d78e875312aa528a98e |
|
BLAKE2b-256 | c25dd125a801e5daf47c240a693db4f23f6e7336fee1879a1721fc1d7a266206 |