so am I doing it correctly?
also, is there any way for me to get or print the current learning rate during the training process?
for example:
1 epoch: learning rate - 2e-07
2 epoch: learning rate - 1e-0…
…