traced_hp

模块用于指定具有跟踪功能的基本超参数。

TracedHpRegistry

一个简单的注册表,用于跟踪不同的追踪hp类及其符号。

TracedHp

一个hparam,展示了处理跟踪所需的附加功能。

class TracedHp

基础类: Hparam

一个hparam,展示了处理跟踪所需的附加功能。

classmethod initialize_from(hp)

从现有的普通 Trace 初始化一个新的 hparam。

Parameters:

hp (Hparam) –

Return type:

TracedHp

resolve_dependencies(sym, get_hp)

通过符号映射解析hparam的依赖关系。

此方法遍历sym中描述的依赖关系图,并根据当前分配的hparam和父符号生成适当的hparam。

Parameters:
  • sym (符号) – 与自身关联的符号,我们希望为其解析依赖关系。

  • get_hp (Callable[[Symbol], TracedHp]) – 一个返回与符号关联的hparam的函数。

Returns:

描述应与每个符号关联的hparam的映射。

Return type:

Dict[Symbol, TracedHp]

class TracedHpRegistry

基础类:object

一个简单的注册表,用于跟踪不同的追踪hp类及其符号。

classmethod get(sym)

获取与符号关联的Hparam类型。

Parameters:

sym (符号) –

Return type:

类型[TracedHp] |

classmethod initialize_from(sym, hp)

从普通的hparam初始化sym-appropriate hparam。

Parameters:
Return type:

TracedHp

classmethod register(sym_cls)

使用此功能为提供的符号类注册一个新的跟踪hparam类。

用法:

@TracedHpRegistry.register(MySymbol)
class MyHparam(TracedHp):
    pass
Parameters:

sym_cls (类型[符号]) –

Return type:

可调用[[类型[TracedHp]], 类型[TracedHp]]

classmethod unregister(sym_cls)

取消注册先前注册的符号类。

如果未注册hparam类,则会抛出KeyError。

Parameters:

sym_cls (类型[符号]) –

Return type: