[utils.checkpointer] Refactor checkpoint saving and loading

src/ptbench/data/datamodule.py

src/ptbench/engine/trainer.py

src/ptbench/scripts/experiment.py

src/ptbench/scripts/train.py

src/ptbench/utils/checkpointer.py

tests/test_cli.py