Some models and trainer scripts created using pytorch-lightning.
Dependencies will be pulled in automatically by pip
.
To install this package run the following command within this directory:
pip install .
Example usage is in examples/
.
Running python examples/mlp_example.py
and then
python examples/mlp_visualization.py
gives:
Running python examples/gp_example.py
and then
python examples/gp_visualization.py
gives:
To uninstall the package:
pip uninstall pl-utils
To cite acxz/pl-utils
in publications use:
acxz. (2019, December 17). acxz/nn-utils-pytorch: Utilities for Training Neural Networks in Pytorch (Version v0.9.0). Zenodo. http://doi.org/10.5281/zenodo.3579627
Here is a BibTex entry for convenience:
@software{acxz_2019_3579627,
author = {acxz},
title = {{acxz/nn-utils-pytorch: Utilities for Training
Neural Networks in Pytorch}},
month = dec,
year = 2019,
publisher = {Zenodo},
version = {v0.9.0},
doi = {10.5281/zenodo.3579627},
url = {https://doi.org/10.5281/zenodo.3579627}
}
Click here or the above DOI badge for more information.