Closed flufy3d closed 1 month ago
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
改type 就可以加载不同模型 分辨率也可以改 测试了下 224x224 彩色图 DQN 0.1s 每帧 224x224 彩色图 DDQN 0.2s 每帧
我显卡2080ti