juntang-zhuang / Adabelief-Optimizer

Is there a way to suppress these messages by setting some parameters explicitly when they are enabled?

Weight decoupling enabled in AdaBelief
Rectification enabled in AdaBelief

I skimmed through the code and did not notice there is any parameter that we do so. I apologize if I have overlooked any part of the code/documentation. Thank you in advance for your reply.

Environment

adabelief_pytorch 0.2.1
Python 3.8.10

Sorry, these two messages are not allowed to be disabled in the current implementation. Maybe you can consider putting the print function

Adabelief-Optimizer/pypi_packages/adabelief_pytorch0.2.1/adabelief_pytorch/AdaBelief.py

Line 93 in 2855178

if self.weight_decouple:

inside the if condition here and disabling it.

Adabelief-Optimizer/pypi_packages/adabelief_pytorch0.2.1/adabelief_pytorch/AdaBelief.py

Line 47 in 2855178

if print_change_log:

Suppressing weight decoupling and rectification messages