[scripts] Fix usage docs

src/ptbench/scripts/saliency_completeness.py

@@ -25,7 +25,7 @@ logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s")
 
    .. code:: sh
 
-      ptbench saliency-completeness -vv pasa tbx11k-v1-healthy-vs-atb --device="cuda" --weight=path/to/model_final.pth --output-folder=path/to/completeness-scores/
+      ptbench saliency-completeness -vv pasa tbx11k-v1-healthy-vs-atb --device="cuda" --weight=path/to/model_final.ckpt --output-folder=path/to/completeness-scores/
 
 """,
 )

src/ptbench/scripts/train.py

@@ -257,10 +257,10 @@ def train(
     """Trains an CNN to perform image classification.
 
     Training is performed for a configurable number of epochs, and
-    generates at least a final_model.pth.  It may also generate a number
-    of intermediate checkpoints.  Checkpoints are model files (.pth
-    files) that are stored during the training and useful to resume the
-    procedure in case it stops abruptly.
+    generates at least a final_model.ckpt.  It may also generate a
+    number of intermediate checkpoints.  Checkpoints are model files
+    (.ckpt files) that are stored during the training and useful to
+    resume the procedure in case it stops abruptly.
     """
 
     import torch