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.17.tar.gz
(38.6 kB
view hashes)
Built Distribution
fannypack-0.0.17-py3-none-any.whl
(51.6 kB
view hashes)
Close
Hashes for fannypack-0.0.17-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4b4ecd3b63ba74a340a1f56563c5ad3b99871c9996923073473a411ab23394b5 |
|
MD5 | 76d5d0eca659e2e14889e93c4b512e8e |
|
BLAKE2b-256 | 4ae40c882d8f76aa5f7acc86bc271d09d3d07aa0c9e0ca8da3b53708cfdd9d56 |