1

系统信息

  • 操作系统平台和发行版:MacOS
  • TensorFlow 安装自:Colab
  • TensorFlow 版本:2.5.0
  • Python版本:python 3.7
  • GPU型号和内存:Tesla T4

错误 TypeError: Input 'y' of 'Sub' Op has type float16 that does not match type float32 of argument 'x'

当前行为

在使用混合精度并构建 Keras 功能 API 模型 (EfficientNet B0) 时,它显示以下错误 TypeError: Input 'y' of 'Sub' Op has type float16 that does not match type float32 of argument 'x'

注意:由于某种原因 StackOverflow 不允许我发布图片。访问以下链接以更深入地了解该问题: https ://github.com/tensorflow/tensorflow/issues/49725

预期行为

我在上一个单元格中设置的全局策略是mixed_float16. 运行时问题正常,tensorflow 2.4.1因此错误与tensorflow 2.5.0

您可以使用以下笔记本重现相同的错误: https ://colab.research.google.com/drive/1TfNZSIJ_I7IZI35RsGFnTdj-6beMHV2_?usp=sharing

4

0 回答 0