pl-utils

Some models and trainer scripts created using pytorch-lightning.

Dependencies

Dependencies will be pulled in automatically by pip.

Installation

To install this package run the following command within this directory:

pip install .

Usage

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:

Uninstall

To uninstall the package:

pip uninstall pl-utils

Citation

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.