diff --git a/codes/models/networks.py b/codes/models/networks.py index 4f748817..5ddb8818 100644 --- a/codes/models/networks.py +++ b/codes/models/networks.py @@ -59,9 +59,6 @@ def define_G(opt, net_key='network_G', scale=None): xforms = opt_net['num_transforms'] if 'num_transforms' in opt_net.keys() else 8 netG = spsr.SwitchedSpsrWithRef2(in_nc=3, out_nc=3, nf=opt_net['nf'], xforms=xforms, upscale=opt_net['scale'], init_temperature=opt_net['temperature'] if 'temperature' in opt_net.keys() else 10) - elif which_model == "csnln": - import model.csnln as csnln - netG = csnln.CSNLN(munchify(opt_net)) else: raise NotImplementedError('Generator model [{:s}] not recognized'.format(which_model)) diff --git a/codes/requirements.txt b/codes/requirements.txt index 15125658..4d60a0f5 100644 --- a/codes/requirements.txt +++ b/codes/requirements.txt @@ -8,3 +8,4 @@ scp tqdm matplotlib scipy +munch