Amazon web services Sagemaker上的自定义训练Tensorflow:WARNING keras.py:603无法使用tf.function在向前传球时保存层值

Amazon web services Sagemaker上的自定义训练Tensorflow:WARNING keras.py:603无法使用tf.function在向前传球时保存层值,amazon-web-services,tensorflow,keras,batch-processing,amazon-sagemaker,Amazon Web Services,Tensorflow,Keras,Batch Processing,Amazon Sagemaker,我正在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 我不确定这是否值得担心。有人对

我正在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

我不确定这是否值得担心。有人对此有什么想法吗?

此警告记录在2020-12-08的smdebug/tensorflow/keras.py中


从2020年12月23日起,该警告被删除,因此作者显然认为该警告是多余的。

谢谢您提供的详细信息!