Fix GPU setting for systems without GPU

Kirill Kolodiazhnyi · Kirill Kolodiazhnyi · commit 197858e22f2b · 2020-03-21T17:31:51.000+02:00
diff --git a/code2seq.py b/code2seq.py
@@ -8,8 +8,9 @@
 
 if __name__ == '__main__':
     physical_devices = tf.config.list_physical_devices('GPU')
-    tf.config.experimental.set_memory_growth(physical_devices[0], True)
-    # tf.config.set_visible_devices([], 'GPU')
+    if len(physical_devices):
+        tf.config.experimental.set_memory_growth(physical_devices[0], True)
+        # tf.config.set_visible_devices([], 'GPU')
 
     args = read_args()
 
