| 63 | |
| 64 | |
| 65 | def callback_init(cfg): |
| 66 | monitor = cfg.checkpoints["monitor"] |
| 67 | filename = cfg.checkpoints["filename"] |
| 68 | cfg.checkpoints["monitor"] = monitor + "_out" |
| 69 | cfg.checkpoints["filename"] = filename + "_out" |
| 70 | checkpoint_callback_out = instantiate(cfg.checkpoints) |
| 71 | cfg.checkpoints["monitor"] = monitor + "_in" |
| 72 | cfg.checkpoints["filename"] = filename + "_in" |
| 73 | checkpoint_callback_in = instantiate(cfg.checkpoints) |
| 74 | progress_bar = instantiate(cfg.progress_bar) |
| 75 | lr_monitor = LearningRateMonitor() |
| 76 | callbacks = [checkpoint_callback_out, checkpoint_callback_in, progress_bar, lr_monitor] |
| 77 | return callbacks |
| 78 | |
| 79 | |
| 80 | def init_datamodule(cfg): |