issues on AdaBlief-tensorflow about adabelief-optimizer

[dusk666]

python version：3.7.6
tensoflow version：tensorflow-gpu 2.2.0
adabelief-tf ：0.1.0
model using tf.keras.api

[louislbc]

Hi, does the adabelief-tf support tensorflow 1, for instance, tensorflow-gpu 1.14? I found the adabelief-tf uninstalled my tensorflow 1 and installed tensorflow 2. Thanks in advance.

[dusk666]

Hi, does the adabelief-tf support tensorflow 1, for instance, tensorflow-gpu 1.14? I found the adabelief-tf uninstalled my tensorflow 1 and installed tensorflow 2. Thanks in advance.

I think u need tensorflow version>=2.1.0 , because neither adabelief-tf 0.0.1 nor 0.1.0 work in my tensorflow 2.0.0 env.
BUT we can run it in tensorflow 2.2 or 2.3.

[cryu854]

@dusk666 The message is to guide you to set the default arguments for AdaBelief, popping up once you create AdaBeliefOptimizer.

For version, the newest pypi_package cache file of adabelief_tf0.1.0 is not update yet.
Currently, please use the following commands to install:

git clone https://github.com/juntang-zhuang/Adabelief-Optimizer.git
pip install ./Adabelief-Optimizer/pypi_packages/adabelief_tf0.1.0

[cryu854]

@louislbc Sorry, the package currently only supports tensorflow 2.

[juntang-zhuang]

@louislbc @dusk666 We have released version 0.2.0, please try the new version by pip install adabelief-tf==0.2.0, this version should be slightly faster than 0.1.0.

[juntang-zhuang]

@louislbc Please see https://github.com/juntang-zhuang/Adabelief-Optimizer/tree/update_0.3.0/pypi_packages/adabelief_tf0.3.0, there are two subfolders, one for tf1 the other for tf2. It's not pushed to pip yet, so please download the source and try. Thanks for the implementation by @cryu854.