Преобразование модели Keras в Pytorch

Question

У меня есть следующий код в Keras, и я хочу передать его Pytorch. Вы можете помочь мне? Я пытаюсь воспроизвести сеть Pix2pix cGAN. Код, который я прикрепил, должен быть в состоянии создать составную модель.

    # Build the generator

    self.generator = self.build_generator()

    # Input images and their conditioning images

    file_A = Input(shape=self.file_shape)
    file_B = Input(shape=self.file_shape)

    # By conditioning on B generate a fake version of A

    fake_A = self.generator(file_B)

    # For the combined model we will only train the generator

    self.discriminator.trainable = False

    # Discriminators determines validity of translated images / condition pairs

    valid = self.discriminator([fake_A, file_B])

    self.combined = Model(inputs=[file_A, file_B], outputs=[valid, fake_A])
    self.combined.compile(loss=['mse', 'mae'],
                          loss_weights=[1, 100],
                          optimizer=optimizer)