Skip to content

Commit 0e3aefd

Browse files
committed
vram growth = True
1 parent f0a00bb commit 0e3aefd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/params.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@
9898
"""TensorFlow random seed.""")
9999
tf.flags.DEFINE_boolean('log_device_placement', False,
100100
"""Whether to log device placement.""")
101-
tf.flags.DEFINE_boolean('allow_vram_growth', False,
101+
tf.flags.DEFINE_boolean('allow_vram_growth', True,
102102
"""Allow TensorFlow to allocate VRAM as needed,
103103
as opposed to allocating the whole VRAM at program start.""")
104104

0 commit comments

Comments
 (0)