8 lines
404 B
Python
8 lines
404 B
Python
from baselines.deepq import models # noqa
|
|
from baselines.deepq.build_graph import build_act, build_train # noqa
|
|
from baselines.deepq.simple import learn, load # noqa
|
|
from baselines.deepq.replay_buffer import ReplayBuffer, PrioritizedReplayBuffer # noqa
|
|
|
|
def wrap_atari_dqn(env):
|
|
from baselines.common.atari_wrappers import wrap_deepmind
|
|
return wrap_deepmind(env, frame_stack=True, scale=True) |