amazon-web-services - Sagemaker 上的自定义训练 Tensorflow：警告 keras.py:603 在使用 tf.function 前向传递期间无法保存层值

Question

我正在使用 Sagemaker 上的 Tensorflow 训练具有自定义层的自定义模型。它在本地运行顺利，没有错误，没有警告。当我在 Sagemaker 上运行它时，它会产生许多警告，都是一样的：

2021-01-22 15:20:26.660 xxxx.xxxx.compute.internal:32 WARNING keras.py:603] cannot save layer values during forward pass with tf.function

我不确定这是否值得担心。有人对此有任何想法吗？

Answer 1

此警告从 2020 年 12 月 8 日的变更集 9ad5d8251dfa3cab2be494cd49f06678a6a84a4e 第 603 行记录到 smdebug/tensorflow/ keras.py。

在变更集 da3b4e30b09dda4813c7b6b3bef0e7fb0c8243cf 中，从 2020-12-23 的第 604 行中删除了警告，因此显然作者认为警告是多余的。