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)