RAdam implementation on Keras
Link to the oiginal paper: https://arxiv.org/pdf/1908.03265v1.pdf
- Required Libraries
- Installation
- Using
- Keras
- TensorFlow (another backend can be used)
git clone https://github.com/float256/rectified-adam-keras.git
cd rectified-adam-keras
pip3 install .
from radam import RAdam
from keras.models import Sequential
model = Sequential()
...
model.compile(optimizer=RAdam(1e-4), ...)