How to reduce memory size of pygame array to allow passing to Stable_baselines3 algorithm

python pygame reinforcement-learning openai stable-baselines