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