Skip to content

Instantly share code, notes, and snippets.

View ZohebAbai's full-sized avatar
🐢
Data. Train. Test. Deploy.

Zoheb Abai ZohebAbai

🐢
Data. Train. Test. Deploy.
  • Applied Materials
  • Bangalore, India
View GitHub Profile
@ZohebAbai
ZohebAbai / lr_finder.py
Created May 12, 2019 11:31 — forked from jeremyjordan/lr_finder.py
Keras Callback for finding the optimal range of learning rates
import matplotlib.pyplot as plt
import keras.backend as K
from keras.callbacks import Callback
class LRFinder(Callback):
'''
A simple callback for finding the optimal learning rate range for your model + dataset.