add epsilon to parameter

martinsbruveris / tensorflow-image-models

TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.

https://tfimm.readthedocs.io/en/latest/

Apache License 2.0

286 stars 25 forks source link

Closed hyenal closed 2 years ago

hyenal commented 2 years ago

Small PR to add the epsilon parameter for Adam and RMSProp. This parameter is usually useful to play with when using mixed precision or GANs

martinsbruveris commented 2 years ago

Thanks for the PR! Will merge.