https://github.com/beringresearch/lrfinder

Learning Rate Finder using Tensorflow Dataset

https://github.com/beringresearch/lrfinder

Science Score: 10.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
  • codemeta.json file
  • .zenodo.json file
  • DOI references
  • Academic publication links
    Links to: arxiv.org
  • Academic email domains
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (8.8%) to scientific vocabulary

Keywords

keras machine-learning optimization tensorflow
Last synced: 10 months ago · JSON representation

Repository

Learning Rate Finder using Tensorflow Dataset

Basic Info
  • Host: GitHub
  • Owner: beringresearch
  • Language: Python
  • Default Branch: master
  • Homepage:
  • Size: 243 KB
Statistics
  • Stars: 8
  • Watchers: 4
  • Forks: 3
  • Open Issues: 1
  • Releases: 0
Topics
keras machine-learning optimization tensorflow
Created almost 6 years ago · Last pushed almost 6 years ago

https://github.com/beringresearch/lrfinder/blob/master/

# Implementation of the LR Range test

Learning Rate is an important tunable hyperparameter that affects model performance. This repository is a `tf.keras` implementation of the learning rate range test described in [Cyclical Learning Rates for Training Neural Networks](https://arxiv.org/abs/1506.01186) by Leslie N. Smith and uses the [Tensorflow Dataset class](https://www.tensorflow.org/api_docs/python/tf/data/Dataset) to represent potentially large data collections.

## Rationalle

The learning rate range test is a test that provides valuable information about the optimal learning rate. During a pre-training run, the learning rate is increased linearly or exponentially between two boundaries. The low initial learning rate allows the network to start converging and as the learning rate is increased it will eventually be too large and the network will diverge. Typically, a good static learning rate can be found half-way on the descending loss curve.

## Installation

```
git clone https://github.com/beringresearch/lrfinder/
cd lrfinder
python3 -m pip install --editable .
```

## Usage

For usage illustration, refer to the [Jupyter Notebook example](https://github.com/beringresearch/lrfinder/blob/master/examples/example.ipynb)

## References

This code is largely based on:

* [Keras LR Finder](https://github.com/surmenok/keras_lr_finder)
* [Cyclical Learning Rates for Training Neural Networks](https://arxiv.org/abs/1506.01186)

Owner

  • Name: beringresearch
  • Login: beringresearch
  • Kind: user
  • Company: Bering Limited

GitHub Events

Total
Last Year