diff --git a/src/ptbench/scripts/train.py b/src/ptbench/scripts/train.py index b6ef37bcade2708b5507eef5f31f27b8eee25bc7..6003eaffbdd14960e0b4298e366a991c4dd33324 100644 --- a/src/ptbench/scripts/train.py +++ b/src/ptbench/scripts/train.py @@ -181,7 +181,7 @@ def set_reproducible_cuda(): help='A string indicating the accelerator to use (e.g. "auto", "cpu" or "gpu"). If auto, will select the best one available', show_default=True, required=True, - default="auto", + default="cpu", cls=ResourceOption, ) @click.option(