Imported from https://github.com/surmenok/keras_lr_finder