Brain segmentation was not working without pytorch installed

Thiago Franco de Moraes
1 parent 4993876e
Showing 1 changed file with 1 additions and 0 deletions Show diff stats
invesalius/gui/brain_seg_dialog.py
@@ -21,6 +21,7 @@ from invesalius.gui import dialogs
 HAS_THEANO = bool(importlib.util.find_spec("theano"))
 HAS_PLAIDML = bool(importlib.util.find_spec("plaidml"))
 PLAIDML_DEVICES = {}
+TORCH_DEVICES = {}
  
 try:
     import torch
...	...	@@ -21,6 +21,7 @@ from invesalius.gui import dialogs
21	21	HAS_THEANO = bool(importlib.util.find_spec("theano"))
22	22	HAS_PLAIDML = bool(importlib.util.find_spec("plaidml"))
23	23	PLAIDML_DEVICES = {}
	24	+TORCH_DEVICES = {}
24	25
25	26	try:
26	27	import torch
...	...