def __init__(self,
game,
batch_size,
warp_size=(84, 84),
color_pallete=None,
frameskip=4,
name=None):
assert game in games, "{} is not part of the available Atari suite".format(
game)
if color_pallete is None:
color_pallete = grayscale_palette
self.game = game
self.batch_size = batch_size
self.obs_variable = None
self.warp_size = warp_size
self.color_pallete = color_pallete
self.frameskip = frameskip
rom = os.path.join(os.path.dirname(__file__),
'atari-py/atari_py/atari_roms/{}.bin'.format(game))
print("ROM path: %s" % rom)
with tf.variable_scope(name, default_name='AtariInstance'):
self.instances = gym_tensorflow_module.atari_make(
batch_size=batch_size, game=rom)
print("Create Atari instances", self.instances)
def __init__(self,
game,
batch_size,
warp_size=(84, 84),
color_pallete=None,
frameskip=4,
name=None):
"""
:param str game: ゲーム名
:param int batch_size: 何ゲーム同時に扱うか
:param tuple[int,int] warp_size: 観測するときの画面サイズ
:param color_pallete:
:param int frameskip: 1回のアクションを繰り返す回数？
:param str name: Variable Scopeに使う名前
"""
assert game in games, "{} is not part of the available Atari suite".format(
game)
if color_pallete is None:
color_pallete = grayscale_palette
self.game = game
self.batch_size = batch_size
self.obs_variable = None
self.warp_size = warp_size
self.color_pallete = color_pallete
self.frameskip = frameskip
rom = os.path.join(os.path.dirname(__file__), '..',
'atari-py/atari_py/atari_roms/{}.bin'.format(game))
with tf.variable_scope(name, default_name='AtariInstance'):
self.instances = gym_tensorflow_module.atari_make(
batch_size=batch_size, game=rom)