siweic0818
siweic0818
> All you need to change is get_gan_network() to be - > `def get_gan_network(discriminator, shape, generator, optimizer): > > ``` > gan_input = Input(shape=shape) > > x = generator(gan_input) >...
You need to use some specific tf+keras versions to successfully load ManTraNet weights. Versions that work for me: tf 1.8.0 + keras 2.2.0 (the same as ManTraNet github), tf 1.10.0...
Did you use model = keras.utils.multi_gpu_model( model, 2)? The saved weights are created by multiple gpus so if you are using single gpu then the weights won't load.