From 7e839421334fdb7a3cd11ae97049d98ca78e905a Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 24 Oct 2023 15:16:37 +0200
Subject: [PATCH] [scripts] Fix usage docs
---
src/ptbench/scripts/saliency_completeness.py | 2 +-
src/ptbench/scripts/train.py | 8 ++++----
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/ptbench/scripts/saliency_completeness.py b/src/ptbench/scripts/saliency_completeness.py
index f17ff648..2d1de183 100644
--- a/src/ptbench/scripts/saliency_completeness.py
+++ b/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/
""",
)
diff --git a/src/ptbench/scripts/train.py b/src/ptbench/scripts/train.py
index 6eb91fa1..c5c2a3ac 100644
--- a/src/ptbench/scripts/train.py
+++ b/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
--
GitLab