Судя по примерам на веб-сайте TensorFlow: https://github.com/tensorflow/docs/blob/r1.15/site/en/guide/distribute_strategy.ipynb, похоже, что нет ресурсов о том, как адаптировать ваш код для использования стратегии распространения. Мой оригинальный код включает в себя манипулирование тензорами, например tf.expand_dims(x, axis=1)
. Однако, когда используется стратегия распространения, я получил вышеупомянутую ошибку, так как expand_dims()
не может работать с PerReplica
объектом. Подробнее об ошибке ниже:
Содержимое: PerReplica: {0 / реплика: 0 / задача: 0 / устройство: GPU: 0: Tensor ("IteratorGetNext: 0", shape = (?, 2, 3), dtype = float32, устройство = / реплика: 0 / задача: 0 / устройство: GPU: 0), 1 / реплика: 0 / задача: 0 / устройство: GPU: 1: Tensor («IteratorGetNext_1: 0» , shape = (?, 2, 3), dtype = float32, device = / replica: 0 / task: 0 / device: GPU: 1)}
Любой имеет какое-либо представление о любом решении это? * * 1013