В моей установке я тренирую плотный слой керас, где веса слоя становятся вложениями изображения.Затем я беру веса этого слоя и делаю точечное произведение этого с набором входных векторов.
Я хочу запустить это с использованием графических процессоров, и с 1 графическим процессором все работает нормально.Однако я заметил, что при следовании документации для нескольких графических процессоров в режиме параллелизма данных, появляющаяся сводка модели показывает изменение в структуре моей сети.Похоже, что это потому, что он «реплицирует целевую модель один раз на каждое устройство и использует каждую реплику для обработки различной доли входных данных».
Вопрос: Когда яизвлекая веса из моего слоя встраивания, он все равно даст мне весь слой, или мне нужно будет извлечь несколько наборов весов и каким-то образом объединить их?