orbax uses logging.info directly and not the supplied log function, hence impossible to suppress logging w/o silencing the whole INFO log
Open
checkpoint
type:bug
- Dominant language
- Python
- Stars
- 535
- Forks
- 101
- Avg merge
- 1d 13h
- Merged PRs (30d)
- 23
Description
orbax uses logging.info directly. e.g.:
checkpoint_manager.py:620, composite_checkpoint_handler.py:502, bstract_checkpointer.py:35, async_checkpointer.py:170, checkpoint_manager.py:484, checkpoint_manager.py:1701, checkpoint.py:186, checkpoint_manager.py:801, checkpoint_manager.py:1987, checkpoint_manager.py:1408, sync_checkpointer.py:439, etc
and not the supplied log function, hence impossible to suppress logging w/o silencing the whole INFO log.
Also, the log interface has no level parameter, so it's impossible to control the verbosity.
Contributor guide
Assessment
This issue has not been assessed yet.