From cf8118a85bcf7cac0df33f6be3b5bb9474eed8c4 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 17 Oct 2020 08:32:29 -0600 Subject: [PATCH] Allow recurrence to specified for chainedgen --- codes/models/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/networks.py b/codes/models/networks.py index a6841346..2c232fbc 100644 --- a/codes/models/networks.py +++ b/codes/models/networks.py @@ -126,7 +126,7 @@ def define_G(opt, net_key='network_G', scale=None): elif which_model == 'chained_gen': netG = ChainedEmbeddingGen(depth=opt_net['depth']) elif which_model == 'chained_gen_structured': - netG = ChainedEmbeddingGenWithStructure(depth=opt_net['depth']) + netG = ChainedEmbeddingGenWithStructure(depth=opt_net['depth'], recurrent=opt_net['recurrent'] if 'recurrent' in opt_net.keys() else False) elif which_model == 'chained_gen_structuredr2': netG = ChainedEmbeddingGenWithStructureR2(depth=opt_net['depth']) elif which_model == "flownet2":