Skip to main content
  1. Categories/

TensorFlow

Finding a Learning Rate with Tensorflow 2

·3 mins
Implementing the technique in Tensorflow 2 is straightforward. Start from a low learning rate, increase the learning rate and record the loss. Stop when a very high learning rate is reached. Plot the losses and learning rates choosing a learning rate where the loss is decreasing at a rapid rate.