don't prohibit cpu #1

nanoporetech · Nov 23, 2019 · 4ee06ee · 4ee06ee
1 parent bdfeb69
commit 4ee06ee
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bonito/util.py b/bonito/util.py
@@ -26,7 +26,7 @@
 split_cigar = re.compile(r"(?P<len>\d+)(?P<op>\D+)")
 
 
-def init(seed):
+def init(seed, device):
     """
     Initialise random libs and setup cudnn
 
@@ -35,6 +35,7 @@ def init(seed):
     random.seed(seed)
     np.random.seed(seed)
     torch.manual_seed(seed)
+    if device == "cpu": return
     torch.backends.cudnn.enabled = True
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.benchmark = False