Skip to content
This repository was archived by the owner on Jan 2, 2021. It is now read-only.

Commit 0c9937a

Browse files
authored
Merge pull request #22 from msfeldstein/master
Remove cnmem theano flag.
2 parents 5ef872b + fef84c5 commit 0c9937a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

enhance.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def extend(lst): return itertools.chain(lst, itertools.repeat(lst[-1]))
9696
# Load the underlying deep learning libraries based on the device specified. If you specify THEANO_FLAGS manually,
9797
# the code assumes you know what you are doing and they are not overriden!
9898
os.environ.setdefault('THEANO_FLAGS', 'floatX=float32,device={},force_device=True,allow_gc=True,'\
99-
'print_active_device=False,lib.cnmem=1.0'.format(args.device))
99+
'print_active_device=False'.format(args.device))
100100

101101
# Scientific & Imaging Libraries
102102
import numpy as np

0 commit comments

Comments
 (0)