For example, let\'s say I want to use Softmax with temperature after a layer and I also want to decrease the temperature after 5 epochs.
for epoch in range(EPOCHS