本文整理汇总了Python中tensor2tensor.utils.registry.register_problem方法的典型用法代码示例。如果您正苦于以下问题:Python registry.register_problem方法的具体用法?Python registry.register_problem怎么用?Python registry.register_problem使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensor2tensor.utils.registry
的用法示例。
在下文中一共展示了registry.register_problem方法的4个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: register_game
# 需要导入模块: from tensor2tensor.utils import registry [as 别名]
# 或者: from tensor2tensor.utils.registry import register_problem [as 别名]
def register_game(game_name, game_mode="NoFrameskip-v4"):
"""Create and register problems for the game.
Args:
game_name: str, one of the games in ATARI_GAMES, e.g. "bank_heist".
game_mode: the frame skip and sticky keys config.
Raises:
ValueError: if game_name or game_mode are wrong.
"""
if game_name not in ATARI_GAMES:
raise ValueError("Game %s not in ATARI_GAMES" % game_name)
if game_mode not in ATARI_GAME_MODES:
raise ValueError("Unknown ATARI game mode: %s." % game_mode)
camel_game_name = misc_utils.snakecase_to_camelcase(game_name) + game_mode
# Create and register the Problem
cls = type("Gym%sRandom" % camel_game_name,
(T2TGymEnv,), {"base_env_name": camel_game_name})
registry.register_problem(cls)
# Register the atari games with all of the possible modes.
示例2: register_game
# 需要导入模块: from tensor2tensor.utils import registry [as 别名]
# 或者: from tensor2tensor.utils.registry import register_problem [as 别名]
def register_game(game_name, game_mode="Deterministic-v4"):
"""Create and register problems for the game.
Args:
game_name: str, one of the games in ATARI_GAMES, e.g. "bank_heist".
game_mode: the frame skip and sticky keys config.
Raises:
ValueError: if game_name or game_mode are wrong.
"""
if game_name not in ATARI_GAMES:
raise ValueError("Game %s not in ATARI_GAMES" % game_name)
if game_mode not in ATARI_GAME_MODES:
raise ValueError("Unknown ATARI game mode: %s." % game_mode)
camel_game_name = camel_case_name(game_name) + game_mode
# Create and register the Problem
cls = type("Gym%sRandom" % camel_game_name,
(T2TGymEnv,), {"base_env_name": camel_game_name})
registry.register_problem(cls)
# Register the atari games with all of the possible modes.
示例3: _register_babi_problems
# 需要导入模块: from tensor2tensor.utils import registry [as 别名]
# 或者: from tensor2tensor.utils.registry import register_problem [as 别名]
def _register_babi_problems():
"""It dynamically instantiates a class for each babi subsets-tasks.
@registry.register_problem
class BabiQaConcatAllTasks_10k(EditSequenceRegexProblem):
@property
def babi_task_id(self):
return 'qa0'
@property
def babi_subset(self):
return 'en-10k'
It does not put the classes into the global namespace, so to access the class
we rely on the registry or this module's REGISTERED_PROBLEMS list.
It will be available as
registry.problem('babi_qa_concat_all_tasks_10k')
i.e., change camel case to snake case. Numbers are considered lower case
characters for these purposes.
"""
for (subset, subset_suffix) in [('en', '_1k'), ('en-10k', '_10k')]:
for problem_name, babi_task_id in six.iteritems(_problems_to_register()):
problem_class = type('BabiQaConcat' + problem_name + subset_suffix,
(BabiQaConcat,), {
'babi_task_id': babi_task_id,
'babi_subset': subset
})
registry.register_problem(problem_class)
REGISTERED_PROBLEMS.append(problem_class.name)
示例4: _register_babi_problems
# 需要导入模块: from tensor2tensor.utils import registry [as 别名]
# 或者: from tensor2tensor.utils.registry import register_problem [as 别名]
def _register_babi_problems():
"""It dynamically instantiates a class for each babi subsets-tasks.
@registry.register_problem
class BabiQaConcatAllTasks_10k(EditSequenceRegexProblem):
@property
def babi_task_id(self):
return "qa0"
@property
def babi_subset(self):
return "en-10k"
It does not put the classes into the global namespace, so to access the class
we rely on the registry or this module"s REGISTERED_PROBLEMS list.
It will be available as
registry.problem("babi_qa_concat_all_tasks_10k")
i.e., change camel case to snake case. Numbers are considered lower case
characters for these purposes.
"""
for (subset, subset_suffix) in [("en", "_1k"), ("en-10k", "_10k")]:
for problem_name, babi_task_id in six.iteritems(_problems_to_register()):
problem_class = type("BabiQaConcat" + problem_name + subset_suffix,
(BabiQaConcat,), {
"babi_task_id": babi_task_id,
"babi_subset": subset
})
registry.register_problem(problem_class)
REGISTERED_PROBLEMS.append(problem_class.name)