fanwuyang/neural-enhance-super-resolution
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove cnmem theano flag since it doesn't work if you're sharing GPU with display.

Browse Source
This commit is contained in:
Michael Feldstein
2016-10-30 19:24:50 -07:00
parent 5ef872b876
commit fef84c5b44
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enhance.py
+1 -1
View File
@@ -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, # 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! # 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,'\ 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 # Scientific & Imaging Libraries
import numpy as np import numpy as np