From a4e335c18b2f91bad298e0fb7cd02deabaf83352 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Wed, 12 Apr 2023 14:58:39 +0200
Subject: [PATCH] Changed default accelerator
---
src/ptbench/scripts/train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/ptbench/scripts/train.py b/src/ptbench/scripts/train.py
index b6ef37bc..6003eaff 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(
--
GitLab