https://github.com/pd-mera/cart-pole-dqn

Try to reproduce basic example of Deep Q Learning (DQN) with Pytorch
https://github.com/pd-mera/cart-pole-dqn

deep-q-learning gymnasium-environment onnx onnxruntime torch

Last synced: 3 months ago
JSON representation

Try to reproduce basic example of Deep Q Learning (DQN) with Pytorch

• Host: GitHub
• URL: https://github.com/pd-mera/cart-pole-dqn
• Owner: PD-Mera
• Created: 2024-04-30T04:23:12.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2024-04-30T07:50:56.000Z (about 1 year ago)
• Last Synced: 2025-01-08T19:46:46.132Z (5 months ago)
• Topics: deep-q-learning, gymnasium-environment, onnx, onnxruntime, torch
• Language: Python
• Homepage:
• Size: 137 KB
• Stars: 2
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Basic Deep Q Learning with Pytorch

An reproduced code from [Reinforcement Learning (DQN) Tutorial](https://pytorch.org/tutorials/intermediate/reinforcement_q_learning.html) to learn about Deep Reinforcement Learning

## How to run

Training and convert model to onnx with `main.py`

Runing onnx model inference with `run.py`

## Result

Visualize training result

![](./assets/training.png)

Visualize environment result 

![](./assets/visualize.gif)

## Reference

- [Reinforcement Learning (DQN) Tutorial](https://pytorch.org/tutorials/intermediate/reinforcement_q_learning.html)

- [boringtutor/makeGifFromGymOpenAi](https://github.com/boringtutor/makeGifFromGymOpenAi)