python - 警告 :tensorflow:AutoGraph could not transform 并将按原样运行

Question

我的问题与this有关和 this一个在这里。我在 Windows 和 Python 3.7.8 和 Tensorflow 2.2.0 上使用 PyCharm:

print (sys.version)
3.7.8 (tags/v3.7.8:4b47a5b6ba, Jun 28 2020, 08:53:46) [MSC v.1916 64 bit (AMD64)]

print(tf.__version__)
2.2.0

当我从这个 colab 运行此代码时教程:

import os

import numpy as np

import matplotlib.pyplot as plt
import tensorflow as tf
import tensorflow_datasets as tfds

(raw_train, raw_validation, raw_test), metadata = tfds.load(
    'cats_vs_dogs',
    split=['train[:80%]', 'train[80%:90%]', 'train[90%:]'],
    with_info=True,
    as_supervised=True,
)
IMG_SIZE = 160 # All images will be resized to 160x160

def format_example(image, label):
  image = tf.cast(image, tf.float32)
  image = (image/127.5) - 1
  image = tf.image.resize(image, (IMG_SIZE, IMG_SIZE))
  return image, label

然后尝试运行这一行:

train = raw_train.map(format_example)

我收到警告:

WARNING:tensorflow:AutoGraph could not transform <function format_example at 0x00000265A2DB4E58> and will run it as-is.
Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.
Cause: Unable to locate the source code of <function format_example at 0x00000265A2DB4E58>. Note that functions defined in certain environments, like the interactive Python shell do not expose their source code. If that is the case, you should to define them in a .py source file. If you are certain the code is graph-compatible, wrap the call using @tf.autograph.do_not_convert. Original error: could not get source code
To silence this warning, decorate the function with @tf.autograph.experimental.do_not_convert

我只有在本地电脑上使用 PyCharm 时才会收到此警告。当我在 colab 中运行它时没有问题。这里有什么问题，这是相关的还是我可以忽略警告？

Answer 1

像 OP 一样，我在 JupyterLab 中收到此警告，但在 Colab 中没有。但这只有在我使用 lambda 时才会发生。

train = raw_train.map(lambda x: x)

如果我使用像 OP 在这里所做的命名函数，警告就会消失。

train = raw_train.map(format_example)

DataSet.map()即使您处于急切模式，也始终构建图形。我认为图形构建器遇到了某种困惑。