hx173149 / C3D-tensorflow

C3D is a modified version of BVLC tensorflow to support 3D ConvNets.
MIT License
588 stars 262 forks source link

how to change the learning rate #38

Open kingrain213 opened 6 years ago

kingrain213 commented 6 years ago

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?

zzy123abc commented 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.

jackietom commented 6 years ago

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)"