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

Merge pull request #22 from msfeldstein/master

Browse Source 
Remove cnmem theano flag.
This commit is contained in:
Alex J. Champandard
2016-10-31 13:29:31 +01:00
committed by GitHub
parent 5ef872b876 fef84c5b44
commit 0c9937a317
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