tf.distribute.Strategy: TypeError: Не удалось преобразовать объект типа Тензор - PullRequest
0 голосов
/ 28 января 2020

Судя по примерам на веб-сайте 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

...