added support for torch model

Julian Kates-Harbeck · Julian Kates-Harbeck · commit 638306bed08e · 2018-04-05T21:50:00.000-04:00
diff --git a/examples/learn.py b/examples/learn.py
@@ -33,7 +33,7 @@
 from plasma.preprocessor.preprocess import Preprocessor, guarantee_preprocessed
 from plasma.models.loader import Loader
 
-if conf['model']['torch']:
+if 'torch' in conf['model'].keys() and conf['model']['torch']:
     from plasma.models.torch_runner import train, make_predictions_and_evaluate_gpu
 elif conf['model']['shallow']:
     from plasma.models.shallow_runner import train, make_predictions_and_evaluate_gpu
