CSNLN changes (removed because it doesnt train well)

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))
 

codes/requirements.txt

@@ -8,3 +8,4 @@ scp
 tqdm
 matplotlib
 scipy
+munch