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