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
main
Michael Feldstein 9 years ago
parent 5ef872b876
commit fef84c5b44
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
enhance.py
Unescape 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

Write Preview
Loading…
Cancel
Save