Skip to content
Snippets Groups Projects
Commit f53b7d95 authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

Cast device number to int

parent c60bd4f0
No related branches found
No related tags found
1 merge request!4Moved code to lightning
Pipeline #72513 failed
......@@ -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"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment