Pong_gymnasium

The solution for the ALE/Pong-v5 gymnasium environment

Install library

pip install gymnasium tensorboard collections onnx onnxscript numpy==1.26.4
pip install gymnasium[atari, accept-rom-licesnse]

Install pytorch: https://pytorch.org/get-started/locally/

Train

python Pong.py train 1000

Test

python Pong.py /path/to/model.pth