У меня есть класс AutoEncoder, где я пытаюсь питать number_test_data с формой (2933314, 600) к кодирующему. Я получаю неудачный копирование ввода тензора от/задание: localhost/replica: 0/задача: 0/устройство: ЦП: от 0 до/задание: localhost/replica: 0/задача: 0/устройство: GPU: 0 Для запуска отливки: DST Tensor не инициализирован. [OP: CAST] Ошибка вне памяти. Как разрешить эту ошибку с помощью метода API набора данных? Любая помощь будет высоко оценена. < /P>
class AutoEncoder(Model):
def _init_(self):
super(AutoEncoder, self)._init_()
#############
## ENCODER ##
#############
self.encoder = tensorflow1.keras.Sequential([
tensorflow1.keras.layers.Dense(512, activation="relu"),
tensorflow1.keras.layers.Dense(256, activation="relu"),
tensorflow1.keras.layers.Dense(128, activation="relu"),
tensorflow1.keras.layers.Dense(64, activation="relu"),
tensorflow1.keras.layers.Dense(32, activation="relu"),
tensorflow1.keras.layers.Dense(16, activation="relu")])
#############
## DECODER ##
#############
self.decoder = tensorflow1.keras.Sequential([
tensorflow1.keras.layers.Dense(32, activation="relu"),
tensorflow1.keras.layers.Dense(64, activation="relu"),
tensorflow1.keras.layers.Dense(128, activation="relu"),
tensorflow1.keras.layers.Dense(256, activation="relu"),
tensorflow1.keras.layers.Dense(512, activation="relu"),
tensorflow1.keras.layers.Dense(600, activation="sigmoid")])
def call(self, x):
encoded = self.encoder(x)
decoded = self.decoder(encoded)
return decoded
model = AutoEncoder()
encoder_out = model.encoder(normal_test_data).numpy()
Подробнее здесь: https://stackoverflow.com/questions/671 ... icecpu0-to
Неудачное копирование ввода тензора с/задание: localhost/replica: 0/Задача: 0/Устройство: ЦП: от 0 до/задание: localhost ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение