提交 619122c4 编写于 作者: X Xin Pan

make DQN runnable

PYTHONPATH=/paddle/build/python python train.py --rom ./rom_files/pong.bin  --alg DuelingDQN
上级 5d172a5f
...@@ -158,7 +158,8 @@ class DuelingDQNModel(object): ...@@ -158,7 +158,8 @@ class DuelingDQNModel(object):
for i, var in enumerate(policy_vars): for i, var in enumerate(policy_vars):
sync_op = fluid.layers.assign(policy_vars[i], target_vars[i]) sync_op = fluid.layers.assign(policy_vars[i], target_vars[i])
sync_ops.append(sync_op) sync_ops.append(sync_op)
sync_program = sync_program.prune(sync_ops) # The prune API is deprecated, please don't use it any more.
sync_program = sync_program._prune(sync_ops)
return sync_program return sync_program
def act(self, state, train_or_test): def act(self, state, train_or_test):
......
...@@ -9,7 +9,7 @@ import gym ...@@ -9,7 +9,7 @@ import gym
from gym import spaces from gym import spaces
from gym.envs.atari.atari_env import ACTION_MEANING from gym.envs.atari.atari_env import ACTION_MEANING
from ale_python_interface import ALEInterface from atari_py import ALEInterface
__all__ = ['AtariPlayer'] __all__ = ['AtariPlayer']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册