Error running training on multiple GPUs

brjathu / deepcaps

Official Implementation of "DeepCaps: Going Deeper with Capsule Networks" paper (CVPR 2019).

MIT License

151 stars 48 forks source link

Closed grische closed 4 years ago

grische commented 4 years ago

When running training on multiple GPUs, it fails on creating the Keras model

AttributeError: '_TfDeviceCaptureOp' object has no attribute '_set_device_from_string'

This seems to be related to TF1.14.0 and Keras<2.3.0: https://github.com/tensorflow/tensorflow/issues/30728