Custom callbacks with warm up and cosine decay in TensorFlow

Question

I referred to this example and especially the callback presented here. So, I decided to write out a callback inspired by this one. Basically, it combines warm-ups and cosine