diff --git a/run.py b/run.py
index 5d3aaa8..83de8ef 100644
--- a/run.py
+++ b/run.py
@@ -9,12 +9,12 @@ from keras import backend as K
 K.clear_session()
 
 gpus = tf.config.experimental.list_physical_devices('GPU')
-if gpus:
-  try:
-    tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=6024)])
-  except RuntimeError as e:
-    print(e)
-    os.exit(1)
+# if gpus:
+#   try:
+#     tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=6024)])
+#   except RuntimeError as e:
+#     print(e)
+#     os.exit(1)
 
 print(tf.config.experimental.list_physical_devices())