diff --git a/src/ptbench/scripts/train.py b/src/ptbench/scripts/train.py index e331108db6ec48aaf42fd3593bc0b0abb379849a..8f2f51c923c176a2929a9c78baf1049b43ed53ee 100644 --- a/src/ptbench/scripts/train.py +++ b/src/ptbench/scripts/train.py @@ -125,7 +125,7 @@ logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s") ) @click.option( "--device", - "-d", + "-x", help='A string indicating the device to use (e.g. "cpu" or "cuda:0")', show_default=True, required=True,