From 4fdb89b9e7b655324ca8899f8e1c1e45a66ddb04 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Fri, 15 Dec 2023 13:36:46 +0100
Subject: [PATCH] [scripts.generate_saliencymaps] Improve cli docs
---
src/ptbench/scripts/generate_saliencymaps.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/ptbench/scripts/generate_saliencymaps.py b/src/ptbench/scripts/generate_saliencymaps.py
index 08cb9d28..3277db96 100644
--- a/src/ptbench/scripts/generate_saliencymaps.py
+++ b/src/ptbench/scripts/generate_saliencymaps.py
@@ -27,7 +27,7 @@ logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s")
.. code:: sh
- ptbench generate-saliencymaps -vv densenet tbx11k-v1-healthy-vs-atb --weight=path/to/model_final.ckpt --output-folder=path/to/output
+ ptbench generate-saliencymaps -vv densenet tbx11k-v1-healthy-vs-atb --weight=path/to/model-at-lowest-validation-loss.ckpt --output-folder=path/to/output
""",
)
--
GitLab