zhangzhang-vvv

Does this code set the learning rate？when i use my own dataset，the loss becomes nan and i want to decrease the learning-rate but i cant find it