diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py index 44b7088c207e8eea6fb04e2370614425ea8246b7..9f28ae68283b3719ddf275f1925bbc979961c1d3 100644 --- a/modules/deeplearning/cloudheight.py +++ b/modules/deeplearning/cloudheight.py @@ -844,7 +844,7 @@ class CloudHeightNN: def run(self, matchup_dict, train_dict=None, valid_dict=None): #with self.strategy.scope(): - with tf.device('/device:GPU:'+self.gpu_device): + with tf.device('/device:GPU:'+str(self.gpu_device)): self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict) self.build_model() self.build_training()