(note)
| 113 | logging.info("\033[33mNOTE\033[0m: " + s, *a) |
| 114 | |
| 115 | def write_note(note): |
| 116 | if jax.process_index() == 0: |
| 117 | info("%s", note) |
| 118 | |
| 119 | write_note("Initializing logging...") |
| 120 | if config.wandb.get("log_wandb", False): |
no test coverage detected