Hey, sorry I haven't run your code to check, but here does this reset the temperature parameter after each epoch?
Seems like it resets in every 100 batch, so not after each epoch.