AdaBound optimizer in Keras
Project description
Keras AdaBound
AdaBound optimizer in Keras.
Install
pip install keras-adabound
Usage
Use the optimizer
from keras_adabound import AdaBound
model.compile(optimizer=AdaBound(lr=1e-3, final_lr=0.1), loss=model_loss)
Load with custom objects
from keras_adabound import AdaBound
model = keras.models.load_model(model_path, custom_objects={'AdaBound': AdaBound})
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
keras-adabound-0.2.0.tar.gz
(4.5 kB
view hashes)