(cfg, logger=None, phase='test', **kwargs)
| 4 | |
| 5 | |
| 6 | def build_callbacks(cfg, logger=None, phase='test', **kwargs): |
| 7 | callbacks = [] |
| 8 | logger = logger |
| 9 | |
| 10 | # Rich Progress Bar |
| 11 | callbacks.append(progressBar()) |
| 12 | |
| 13 | # Checkpoint Callback |
| 14 | if phase == 'train': |
| 15 | callbacks.extend(getCheckpointCallback(cfg, logger=logger, **kwargs)) |
| 16 | |
| 17 | return callbacks |
| 18 | |
| 19 | def getCheckpointCallback(cfg, logger=None, **kwargs): |
| 20 | callbacks = [] |