From fef84c5b44d8b2d0317351f3cc1d09d686ea908c Mon Sep 17 00:00:00 2001 From: Michael Feldstein Date: Sun, 30 Oct 2016 19:24:50 -0700 Subject: [PATCH] Remove cnmem theano flag since it doesn't work if you're sharing GPU with display. --- enhance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhance.py b/enhance.py index 23ca9d2..cb67e17 100755 --- a/enhance.py +++ b/enhance.py @@ -96,7 +96,7 @@ print("""{} {}Super Resolution for images and videos powered by Deep Learning! # Load the underlying deep learning libraries based on the device specified. If you specify THEANO_FLAGS manually, # the code assumes you know what you are doing and they are not overriden! os.environ.setdefault('THEANO_FLAGS', 'floatX=float32,device={},force_device=True,allow_gc=True,'\ - 'print_active_device=False,lib.cnmem=1.0'.format(args.device)) + 'print_active_device=False'.format(args.device)) # Scientific & Imaging Libraries import numpy as np