cuizhongren45 / cartpole_dqn_pytorch

基于gym和pytorch的cartpole训练代码

cartpole_dqn_pytorch

基于gym和pytorch的cartpole训练代码

cartpole-dqn

依赖库：

• torch （pytorch官网上按照教程安装合适自己环境的库）
• gymnasium （gym库的升级版，openai新推出的库用于代替gym）
• numpy （深度强化学习中常用到的库）

文件说明：

• cartpole_dqn.pt 保存的模型,可以用于保持平衡杆的稳定
• dqn_test.py 检测模型效果的文件
• dqn_train.py 训练模型的文件

参考:

主要参考链接如下,对代码做了小改动

https://zhuanlan.zhihu.com/p/117287339

感谢点赞和⭐