【Tensorflow】Kerasを使用して、BN層を含むネットワークの複数GPUトレーニングでエラーが発生します。

2019年7月21日#Tensorflow184

AI 翻訳

この記事はAIを通じて英語から日本語に翻訳されました。原文を表示

AI が生成した要約

【Tensorflow】用Keras多GPU训练含有BN层的网络报错。报错内容为AttributeError: 'DeviceSpec' object has no attribute 'split'。解决方法是修改tensorflow/python/framework/device.py文件中的代码。将一行代码注释掉并替换为另一行代码。修改后保存退出即可正常训练。

Keras で multi_gpu_model を使用する場合、次のエラーメッセージが表示されます：

AttributeError: 'DeviceSpec'オブジェクトに '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 "")

保存して終了すると、正常にトレーニングできます。