Skip to content
Snippets Groups Projects
Commit 841e7aeb authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[scripts] added exit when error with dropout/light options in train DR-GAN on multi-PIE

parent d81b0a07
Branches
Tags
No related merge requests found
......@@ -101,6 +101,7 @@ def main(user_input=None):
if bool(args['--light']):
if dropout:
logger.error("The light architecture does not support dropout - so drop this option ;)")
sys.exit()
from bob.learn.pytorch.architectures import DRGAN_encoder as drgan_encoder
from bob.learn.pytorch.architectures import DRGAN_decoder as drgan_decoder
from bob.learn.pytorch.architectures import DRGAN_discriminator as drgan_discriminator
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment