XR-stb / DQN_WUKONG

基于DQN的黑神话悟空强化AI
54 stars 17 forks source link

大更新 #10

Closed flufy3d closed 1 month ago

flufy3d commented 1 month ago
  1. Context 特征值输入
  2. 彩色图
  3. 多模型支持
  4. 添加DDQN
  5. 配置文件系统
model:
  type: 'DQN'  # Options: 'DQN', 'DDQN', etc.
  model_file: 'model_weight'
  DQN:
    replay_size: 2000
    gamma: 0.9
    initial_epsilon: 0.5
    final_epsilon: 0.009
    lr: 0.001
    batch_size: 32
  DDQN:
    replay_size: 3000
    gamma: 0.95
    initial_epsilon: 0.6
    final_epsilon: 0.01
    lr: 0.0005
    batch_size: 64
training:
  episodes: 3000
  update_step: 200
environment:
  width: 224
  height: 224
  action_size: 6

改type 就可以加载不同模型 分辨率也可以改 测试了下 224x224 彩色图 DQN 0.1s 每帧 224x224 彩色图 DDQN 0.2s 每帧

我显卡2080ti