Updated docstrings

src/ptbench/engine/callbacks.py

src/ptbench/engine/predictor.py

src/ptbench/engine/trainer.py

src/ptbench/scripts/predict.py

src/ptbench/scripts/train.py

src/ptbench/utils/checkpointer.py