akko/diffvg
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix error in render_tensorflow

Browse Source
This commit is contained in:
oeway
2020-09-28 23:23:59 +02:00
parent 5b65ad38b1
commit 08739cdd0a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pydiffvg_tensorflow/render_tensorflow.py
View File

@@ -1,3 +1,4 @@
import os
import tensorflow as tf import tensorflow as tf
import diffvg import diffvg
import pydiffvg_tensorflow as pydiffvg import pydiffvg_tensorflow as pydiffvg
@@ -457,7 +458,7 @@ def render(*x):
The main TensorFlow interface of C++ diffvg. The main TensorFlow interface of C++ diffvg.
""" """
assert(tf.executing_eagerly()) assert(tf.executing_eagerly())
if pydiffvg.get_use_gpu() and os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] != 'true': if pydiffvg.get_use_gpu() and os.environ.get('TF_FORCE_GPU_ALLOW_GROWTH') != 'true':
print('******************** WARNING ********************') print('******************** WARNING ********************')
print('Tensorflow by default allocates all GPU memory,') print('Tensorflow by default allocates all GPU memory,')
print('causing huge amount of page faults when rendering.') print('causing huge amount of page faults when rendering.')