(x, *args, **kargs)
| 44 | raise Exception('Unknown logging backend.') |
| 45 | |
| 46 | def train_log(x, *args, **kargs): |
| 47 | |
| 48 | if train_log_backend == 'print': |
| 49 | print(x) |
| 50 | elif train_log_backend == 'loguru': |
| 51 | loguru.logger.info(x) |
| 52 | elif train_log_backend == 'wandb': |
| 53 | wandb.log(x, *args, **kargs) |
| 54 | else: |
| 55 | raise Exception('Unknown logging backend.') |
| 56 | |
| 57 |
no outgoing calls