Open gunsodo opened 2 years ago
Sorry, these two messages are not allowed to be disabled in the current implementation. Maybe you can consider putting the print function https://github.com/juntang-zhuang/Adabelief-Optimizer/blob/2855178eff6776762793f915ce2c615f776a1e4b/pypi_packages/adabelief_pytorch0.2.1/adabelief_pytorch/AdaBelief.py#L93 inside the if
condition here and disabling it.
https://github.com/juntang-zhuang/Adabelief-Optimizer/blob/2855178eff6776762793f915ce2c615f776a1e4b/pypi_packages/adabelief_pytorch0.2.1/adabelief_pytorch/AdaBelief.py#L47
Is there a way to suppress these messages by setting some parameters explicitly when they are enabled?
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