Commit 9aba833eb13296cee0ce4d9d82ae8e38fff52c07
1 parent
cbbe0f41
Exists in
master
Making it possible to run with only torch
Showing
1 changed file
with
2 additions
and
2 deletions
Show diff stats
invesalius/gui/frame.py
| @@ -790,13 +790,13 @@ class Frame(wx.Frame): | @@ -790,13 +790,13 @@ class Frame(wx.Frame): | ||
| 790 | 790 | ||
| 791 | def OnBrainSegmentation(self): | 791 | def OnBrainSegmentation(self): |
| 792 | from invesalius.gui import brain_seg_dialog | 792 | from invesalius.gui import brain_seg_dialog |
| 793 | - if brain_seg_dialog.HAS_PLAIDML or brain_seg_dialog.HAS_THEANO: | 793 | + if brain_seg_dialog.HAS_PLAIDML or brain_seg_dialog.HAS_THEANO or brain_seg_dialog.HAS_TORCH: |
| 794 | dlg = brain_seg_dialog.BrainSegmenterDialog(self) | 794 | dlg = brain_seg_dialog.BrainSegmenterDialog(self) |
| 795 | dlg.Show() | 795 | dlg.Show() |
| 796 | else: | 796 | else: |
| 797 | dlg = wx.MessageDialog(self, | 797 | dlg = wx.MessageDialog(self, |
| 798 | _("It's not possible to run brain segmenter because your system doesn't have the following modules installed:") \ | 798 | _("It's not possible to run brain segmenter because your system doesn't have the following modules installed:") \ |
| 799 | - + " PlaidML or Theano" , | 799 | + + " Torch, PlaidML or Theano" , |
| 800 | "InVesalius 3 - Brain segmenter", | 800 | "InVesalius 3 - Brain segmenter", |
| 801 | wx.ICON_INFORMATION | wx.OK) | 801 | wx.ICON_INFORMATION | wx.OK) |
| 802 | dlg.ShowModal() | 802 | dlg.ShowModal() |