Is this only works for Feed Forward?

[surmenok]

Is this only works for Feed Forward?

[tarasivashchuk]

This will work for Model object (keras.models.Model or tf.keras.models.Model) as long as the optimizer function used when compiling the model has a "learning_rate" parameter.

Actually, I just checked the code for this repo and it uses "lr", which is technically the same in the Keras source code, but only supported for the sake of backwards compatibility. The recommended usage "learning_rate", and I will refactor that and make a pull request after I'm done with this comment. 😄

So you could even a multi-input model with a CNN, an RNN, and a Feed-Forward network within one model and use this - the model doesn't matter. I would recommend learning a bit more about how this works and the corresponding login.

Here are some resources you could look over:
"How to Use the Learning Rate Finder in TensorFlow" by Ashwath Salimath
Learning Rate Finder Documentation from Fast.ai

Also, I'm not sure how, but this issue should be closed.

[surmenok]

No, it should work for different kinds of neural networks, including CNN. This approach was tried on fully connected networks, CNN and LSTM.