2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Disabled growing of GPU RAM

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-01 17:04:54 +02:00
parent 706caf445a
commit 04a405545f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/main.py
View File

@ -176,7 +176,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
from twomartens.masterthesis import ssd from twomartens.masterthesis import ssd
config = tf.ConfigProto() config = tf.ConfigProto()
config.gpu_options.allow_growth = True config.gpu_options.allow_growth = False
tf.enable_eager_execution(config=config) tf.enable_eager_execution(config=config)
batch_size = 16 batch_size = 16
image_size = 300 image_size = 300