search

keras-team / keras-contrib

Keras community contributions
MIT License
1.58k stars 650 forks source link

LARS optimization #530

Open mmbejani opened 4 years ago

mmbejani commented 4 years ago

I try to create an object from the LARS optimizer, but I got the following error:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/bmn4_bejani/keras-contrib/keras_contrib/optimizers/lars.py", line 46, in __init__
    self.lr = K.variable(lr, name='lr')
AttributeError: can't set attribute

My config: Tensorflow: 2.1 keras: 2.3.1 CUDA: 10.1 cuDnn 7.5