just an embedding test to overfit an image generation model - View it on GitHub