Skip to content

Trainable discriminator

Vladimír Višňovský requested to merge trainable_discriminator into master

Addresses issue #5.

  • Discriminator is now frozen during training. It can be checked via new flag verbose that the discriminator has 0 trainable params.
  • Changed default activation functions for encoder/decoder based on tests made
  • Added more friendly way to visualise structure of models (verbose)
  • Fixed wrong order of neurons in layers
  • Model naming fix

Merge request reports