a90bfb4a2cc8f3a4aed14edc3de701b72d085a32,capsulenet.py,,test,#,141

Before Change

    print()
    print("Reconstructed images are saved to ./real_and_recon.png")
    print("-"*50)
    plt.imshow(plt.imread("real_and_recon.png", ))
    plt.show()


def load_mnist():

After Change

    print()
    print("Reconstructed images are saved to %s/real_and_recon.png" % args.save_dir)
    print("-" * 30 + "End: test" + "-" * 30)
    plt.imshow(plt.imread(args.save_dir + "/real_and_recon.png"))
    plt.show()


def manipulate_latent(model, data, args):

Instances