Не удалось скопировать входной тензор из /job:localhost/replica:0/task:0/device:CPU:0 в /job:localhost/replica:0/task:0/ ⇐ Python

[Anonymous]

У меня есть класс автокодировщика, в котором я пытаюсь передать кодировщику Normal_test_data с формой (2933314, 600). Я получаю входной тензор неудачного копирования из /job:localhost/replica:0/task:0/device:CPU:0 в /job:localhost/replica:0/task:0/device:GPU:0 для запуска Cast : Тензор Dst не инициализирован. [Op:Cast] Ошибка нехватки памяти. Как устранить эту ошибку с помощью метода API набора данных? Любая помощь будет принята с благодарностью.
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