Загрузка модели с контрольной точки не удалась? - PullRequest
1 голос
/ 09 июля 2019

https://www.tensorflow.org/beta/tutorials/generative/dcgan Я следую этому руководству для DCGAN и пытаюсь восстановить модель с сохраненной мной контрольной точки.Но при попытке загрузить модель выдает ошибку:

AssertionError: Nothing except the root object matched a checkpointed value. Typically this means that the checkpoint does not match the Python program. The following objects have no matching checkpointed value: [<tensorflow.python.keras.layers.advanced_activations.LeakyReLU object at 0x7f05b545fc88>, <tf.Variable 'conv2d_transpose_5/kernel:0' shape=(3, 3, 1, 64) dtype=float32, numpy=.......

Я попытался изменить контрольную точку, сохранив только часть генератора, так как это все, что мне нужно здесь.После этого я делаю

latest = tf.train.latest_checkpoint(checkpoint_dir)
gen_mod = make_generator_model() #This is already defined in the code
gen_mod.load_weights(latest)
sample = gen_mod(noise,training=False)

, что дает мне ошибку.Есть ли способ просто загрузить генераторную часть?Я хочу, чтобы у меня была возможность генерировать изображения с моделью генератора из заданной контрольной точки.

...