Отключение тензорного автографа - PullRequest
Новая
       2

Отключение тензорного автографа

0 голосов
/ 19 января 2020

Я использую инструкцию tf.data.Dataset.map(process_fn), функция отображения состоит из чисто тензорных графовых функций, но все же кажется, что Autograph пытается их преобразовать. Как я могу предотвратить это?

Как я могу заставить тензорного потока использовать мою часть кода (которая определяет граф) как есть? 

def process_fn(item):
    assert 'image' in item
    # this should be executed right not every time graph is executed
    image = tf.image.convert_image_dtype(item.pop('image'), tf.float32)
    image = tf.multiply(tf.subtract(image, 0.5), 2)
    return image

По какой-то причине тензор потока хочет преобразовать эту функцию и сообщает предупреждение о невозможности и о том, что оно будет использовано как есть. Вопрос в том, почему вообще делается попытка использовать Автограф? 

W0119 14:55:15.113813 140297917577024 ag_logging.py:146] Entity 
<function geospatial_input.<locals>.process_fn at 0x7f991b5fe280> could 
not be transformed and will be executed as-is. Please report this to 
the AutoGraph team. When filing the bug, set the verbosity to 10 (on 
Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.
...