diff --git a/src/ptbench/scripts/predict.py b/src/ptbench/scripts/predict.py index 0715e423393570d4d63affafeec4c78eb0db28ba..c13a4f242b783335f85072d88a6e951fb988a398 100644 --- a/src/ptbench/scripts/predict.py +++ b/src/ptbench/scripts/predict.py @@ -71,7 +71,7 @@ logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s") requirements for the network).""", required=True, show_default=True, - default=10, + default=1, type=click.IntRange(min=1), cls=ResourceOption, )