AttributeError: 'AdaBeliefOptimizer' object has no attribute '_set_hyper' about adabelief-optimizer

[juntang-zhuang]

No idea with tensorflow, it's contributed by another user. Maybe you can try tensorflow addon https://www.tensorflow.org/addons/api_docs/python/tfa/optimizers/AdaBelief

[juntang-zhuang]

Are you using pytorch or tensorflow version? I don't think I have written or used any function named _set_hyper

[SamMohel]

Tensorflow version with 2.15.0. The exact error is

s~/.local/lib/python3.10/site-packages/adabelief_tf/AdaBelief_tf.py:148, in AdaBeliefOptimizer.__init__(self, learning_rate, beta_1, beta_2, epsilon, weight_decay, rectify, amsgrad, sma_threshold, total_steps, warmup_proportion, min_lr, name, print_change_log, **kwargs)
    145     print(Style.RESET_ALL)
    146 # ------------------------------------------------------------------------------
--> 148 self._set_hyper("learning_rate", kwargs.get("lr", learning_rate))
    149 self._set_hyper("beta_1", beta_1)
    150 self._set_hyper("beta_2", beta_2)

https://github.com/juntang-zhuang/Adabelief-Optimizer/blob/update_0.2.0/pypi_packages/adabelief_tf0.2.1/adabelief_tf/AdaBelief_tf.py#L148

[SamMohel]

Thanks i already used optimizer = optim.AdaBelief(model.parameters(), lr=1e-3) and worked but the accuracy doesn't greater than using Adam optimizer. My task is for classification