作者有没有pytorch版本？

XR-stb / DQN_WUKONG

基于DQN的黑神话悟空强化AI

54 stars 17 forks source link

Open chenzean opened 2 months ago

flufy3d commented 2 months ago

@XR-stb 有必要换到pytorch吗？

XR-stb commented 2 months ago

没有，你找个pytorch的dqn代码改一下就能用。

XR-stb commented 2 months ago

@XR-stb 有必要换到pytorch吗？

没什么区别

flufy3d commented 2 months ago

模型定义说是pytorch 代码简洁些社区也更活跃

flufy3d commented 1 month ago

我搞了个分支升级到了pytorch版本还有些小优化 https://github.com/flufy3d/DQN_WUKONG/tree/pytorch

@XR-stb 很丝滑现在每帧只需要80毫秒了 pytorch的模型文件也小很多就60多兆你可以试试看看要不要换到pytorch

flufy3d commented 1 month ago

你看看要是没问题我就合并到main 然后提交pr给你

XR-stb commented 1 month ago

emm，现在已经全面更换为pytorch了，you can try it!