Open kingrain213 opened 6 years ago
You can use sgd to change the learning rate. This code is adam,so the learning rate is waving only in a little scale.
I think you can change the learning rate by changing the learning rate of opt_stable and opt_finetuning in the trainc3d....py file. In detail, you can change the number in " opt_stable = tf.train.AdamOptimizer(1e-6)" and " opt_finetuning = tf.train.AdamOptimizer(1e-3)"
I have trained my own model use your code, thank you for your share. now the rate is waving only in a little scale, but I havent got the learning rate parameter, could you tell me how can I change the learning rate ? what is your original learning rate parameter in your train?