【Tensorflow】使用Keras，在含有BN層的網絡上進行多GPU訓練時報錯

2019年7月21日#Tensorflow184

AI 翻譯

這篇文章透過AI由英文翻譯成繁體中文。查看原文

AI 生成的摘要

這篇文章提到在使用Keras的multi_gpu_model時，如果訓練含有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 "")

保存退出，就能正常訓練了。