"""Build PixelGen: write the hand-written weights to weights/pixelgen.pt."""

import sys

from pixelgen import weights


def main(argv=None):
    path = weights.save()
    print(f"wrote {path}")
    model = weights.load(path)
    print(f"loaded back: {sum(p.numel() for p in model.buffers())} buffer values")
    return 0


if __name__ == "__main__":
    sys.exit(main())