【Tensorflow】使用Keras，在包含BN层的网络上进行多GPU训练时报错。

2019年7月21日#Tensorflow153

AI 翻译

这篇文章通过AI由英语翻译成中文。查看原文

AI 生成的摘要

这篇文章讨论了在使用Keras进行多GPU训练时，遇到含有BN层的网络报错的问题。文章提供了解决方法，即修改tensorflow/python/framework/device.py文件中的一行代码，保存退出后即可正常训练。

如果在 keras 中使用 multi_gpu_model，报错内容：

AttributeError: 'DeviceSpec' object has no attribute 'split'

如果出现这个错误，多半都是源码造成的。

请查看 tensorflow/python/framework/device.py 这个文件，

查找下面这一行：

current_device = DeviceSpec.from_string(node_def.device or "")

将上面这一行注释掉，改成下面这一行：

current_device = node_def.device if isinstance(node_def.device, DeviceSpec) else DeviceSpec.from_string(node_def.device or "")

保存退出，就能正常训练了。