def load_gan(args):
outdir = args.outdir
modelDict = AttrDict({})
modelDict.discriminator = loadModel(outdir + "/discriminator")
modelDict.gen_disc = loadModel(outdir + "/gen_disc")
modelDict.generator = loadModel(outdir + "/generator")
return modelDict
def load_autoencoder(args):
outdir = args.outdir
modelDict = AttrDict({})
modelDict.ae = loadModel(outdir + "/ae")
modelDict.encoder = loadModel(outdir + "/encoder")
modelDict.generator = loadModel(outdir + "/generator")
return modelDict