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,6 +21,7 @@ from invesalius.gui import dialogs
21	HAS_THEANO = bool(importlib.util.find_spec("theano"))	21	HAS_THEANO = bool(importlib.util.find_spec("theano"))
22	HAS_PLAIDML = bool(importlib.util.find_spec("plaidml"))	22	HAS_PLAIDML = bool(importlib.util.find_spec("plaidml"))
23	PLAIDML_DEVICES = {}	23	PLAIDML_DEVICES = {}
		24	+TORCH_DEVICES = {}
24		25
25	try:	26	try:
26	import torch	27	import torch