atgambardella / pytorch-es

Evolution Strategies in PyTorch
MIT License
349 stars 37 forks source link

Learning rate #5

Closed sahiliitm closed 7 years ago

sahiliitm commented 7 years ago

Is the default learning rate of 0.3 corresponding to 12 threads or 40? Should learning rate scale linearly with batch size?

atgambardella commented 7 years ago

The defaults are assuming that you're training on cartpole, with 40 processes (that being said, most reasonable hyperparameters should work with cartpole). Learning rate should scale in some way with batch size, but most likely not linearly. You'll just have to experiment with what works.

I'll also sometime soon be adding other optimizers, so learning rates will naturally be different for them anyway.