(*args, **kwargs)
| 977 | |
| 978 | # allow checkpointing via USR1 |
| 979 | def melk(*args, **kwargs): |
| 980 | # run all checkpoint hooks |
| 981 | if trainer.global_rank == 0: |
| 982 | print("Summoning checkpoint.") |
| 983 | ckpt_path = os.path.join(ckptdir, "last.ckpt") |
| 984 | trainer.save_checkpoint(ckpt_path) |
| 985 | |
| 986 | def divein(*args, **kwargs): |
| 987 | if trainer.global_rank == 0: |