remove gpu setting

EdwardSmith1884 · Jul 31, 2018 · f883d5f · f883d5f
1 parent 7c15f27
commit f883d5f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/occupancy.py b/occupancy.py
@@ -61,7 +61,6 @@
 config = tf.ConfigProto()
 config.gpu_options.allow_growth = True
 sess=tf.Session()
-tl.ops.set_gpu_fraction(sess=sess, gpu_fraction=0.999)
 sess.run(tf.global_variables_initializer())
 
 ####### load checkpoints and files ###############