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.
main
Alex J. Champandard 9 years ago committed by GitHub
parent 5ef872b876 fef84c5b44
commit 0c9937a317
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