From f53b7d95cdd8831ff770b5b555f6945f0126ccf5 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Mon, 17 Apr 2023 16:42:12 +0200
Subject: [PATCH] Cast device number to int
---
src/ptbench/engine/trainer.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/ptbench/engine/trainer.py b/src/ptbench/engine/trainer.py
index 0db5386c..896ce9a3 100644
--- a/src/ptbench/engine/trainer.py
+++ b/src/ptbench/engine/trainer.py
@@ -41,6 +41,7 @@ class AcceleratorProcessor:
if len(split_accelerator) > 1:
devices = split_accelerator[1:]
+ devices = [int(i) for i in devices]
else:
devices = "auto"
--
GitLab