powered by
This function adjusts the learning rate at each epoch according to a custom schedule.
learn_rate_scheduler(initial_lr, schedule, epochs)
A numeric vector representing the learning rate for each epoch.
A numeric value representing the initial learning rate.
A function that takes the epoch number as input and returns the adjusted learning rate.
An integer specifying the total number of epochs.
schedule <- function(epoch) { 0.01 * (0.9 ^ epoch) } learn_rate_scheduler(0.01, schedule, epochs = 10)
Run the code above in your browser using DataLab