Skip to content

Instantly share code, notes, and snippets.

View WittmannF's full-sized avatar
🇧🇷

Fernando Marcos Wittmann WittmannF

🇧🇷
View GitHub Profile
@WittmannF
WittmannF / lr_finder.py
Last active May 30, 2019 13:38 — 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.