(*args, **kwargs)
| 944 | |
| 945 | # allow checkpointing via USR1 |
| 946 | def melk(*args, **kwargs): |
| 947 | # run all checkpoint hooks |
| 948 | if trainer.global_rank == 0: |
| 949 | print("Summoning checkpoint.") |
| 950 | ckpt_path = os.path.join(ckptdir, "last.ckpt") |
| 951 | trainer.save_checkpoint(ckpt_path) |
| 952 | |
| 953 | def divein(*args, **kwargs): |
| 954 | if trainer.global_rank == 0: |