|
|
|
|
|
|
|
|
|
|
|
|
|
|
@attr.s(auto_attribs=True) |
|
|
|
class PluginSettings: |
|
|
|
plugins: Optional[List[str]] = parser.get_default("plugins") |
|
|
|
|
|
|
|
|
|
|
|
@attr.s(auto_attribs=True) |
|
|
|
class RunOptions(ExportableSettings): |
|
|
|
behaviors: DefaultDict[str, TrainerSettings] = attr.ib( |
|
|
|
factory=lambda: collections.defaultdict(TrainerSettings) |
|
|
|
|
|
|
environment_parameters: Optional[Dict[str, EnvironmentParameterSettings]] = None |
|
|
|
checkpoint_settings: CheckpointSettings = attr.ib(factory=CheckpointSettings) |
|
|
|
plugin_settings: PluginSettings = attr.ib(factory=PluginSettings) |
|
|
|
plugins: List[str] = parser.get_default("plugins") |
|
|
|
cattr.register_structure_hook(PluginSettings, strict_to_cls) |
|
|
|
cattr.register_structure_hook(EngineSettings, strict_to_cls) |
|
|
|
cattr.register_structure_hook(CheckpointSettings, strict_to_cls) |
|
|
|
cattr.register_structure_hook( |
|
|
|