diff options
Diffstat (limited to 'ml_exp/do_ml.py')
-rw-r--r-- | ml_exp/do_ml.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ml_exp/do_ml.py b/ml_exp/do_ml.py index eb57008b0..b37553c24 100644 --- a/ml_exp/do_ml.py +++ b/ml_exp/do_ml.py @@ -121,6 +121,8 @@ def simple_ml(descriptors, Y_pr = tf.tensordot(K_te, alpha, 1) mae = tf.reduce_mean(tf.abs(Y_pr - Y_te)) + else: + raise TypeError('No GPU found, could not create Tensor objects.') else: X_tr = descriptors[:training_size] Y_tr = energies[:training_size] |