diff --git a/src/ptbench/scripts/predict.py b/src/ptbench/scripts/predict.py index 41b10a291e008250747d34a76169de9be57d32fc..21a3ae6636b5392e844a2da27bcec86119dcb45b 100644 --- a/src/ptbench/scripts/predict.py +++ b/src/ptbench/scripts/predict.py @@ -63,7 +63,6 @@ logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s") ) @click.option( "--device", - "-d", help='A string indicating the device to use (e.g. "cpu" or "cuda:0")', show_default=True, required=True, diff --git a/src/ptbench/scripts/train.py b/src/ptbench/scripts/train.py index d487fb66c7c83887dbfc719ffdc92bb4038d535d..d16d74f08a2c5cc8a11e33526a286cd0081d1c62 100644 --- a/src/ptbench/scripts/train.py +++ b/src/ptbench/scripts/train.py @@ -259,7 +259,6 @@ def set_reproducible_cuda(): ) @click.option( "--device", - "-d", help='A string indicating the device to use (e.g. "cpu" or "cuda:0")', show_default=True, required=True,