JianGoForIt / YellowFin

auto-tuning momentum SGD optimizer
Apache License 2.0
422 stars 93 forks source link

AttributeError: 'YFOptimizer' object has no attribute 'compute_gradients' #8

Closed wangxianliang closed 7 years ago

wangxianliang commented 7 years ago

As title.

JianGoForIt commented 7 years ago

Hi @wangxianliang, please check out this

https://github.com/Zehaos/MobileNet/blob/master/optimizer/yellowfin.py

Please keep us updated about whether it works.