search

brendanhasz / probflow

A Python package for building Bayesian models with TensorFlow or PyTorch
http://probflow.readthedocs.io
MIT License
171 stars 17 forks source link

Add adjustable learning rate for PyTorch #53

Closed brendanhasz closed 3 years ago

brendanhasz commented 3 years ago

This allows the learning rate (for the default optimizer) to be adjusted mid-training for PyTorch.

Fixes https://github.com/brendanhasz/probflow/issues/10