From 146ace085935f7e8277921fa161a354f2f6442b5 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 8 Sep 2020 08:04:16 -0600 Subject: [PATCH] CSNLN changes (removed because it doesnt train well) --- codes/models/networks.py | 3 --- codes/requirements.txt | 1 + 2 files changed, 1 insertion(+), 3 deletions(-) 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