nvmidl.utils package
-
handle_amp
(use_amp) → bool
-
handle_determinism
(determinism_config: dict, multi_gpu: bool)
-
handle_tf32
(use_tf32) → bool
-
handle_xla
(use_xla) → bool
-
time_func
(method)
-
class
EvalConfiger
(mmar_root: str, wf_config_file_name=None, env_config_file_name=None, log_config_file_name=None, kv_list=None, base_pkgs=None, module_names=None) Bases:
dlmed.utils.clara_conf.ClaraConfiger
-
finalize_config
(config_ctx: dlmed.utils.wfconf.ConfigContext)
-
process_config_element
(config_ctx: dlmed.utils.wfconf.ConfigContext, node: dlmed.utils.json_scanner.Node)
-
start_config
(config_ctx: dlmed.utils.wfconf.ConfigContext)
-
-
class
TrainConfiger
(mmar_root: str, wf_config_file_name=None, env_config_file_name=None, log_config_file_name=None, kv_list=None, base_pkgs=None, module_names=None) Bases:
dlmed.utils.clara_conf.ClaraConfiger
-
create_component_from_ref
(refs, element)
-
finalize_config
(config_ctx: dlmed.utils.wfconf.ConfigContext)
-
process_config_element
(config_ctx: dlmed.utils.wfconf.ConfigContext, node: dlmed.utils.json_scanner.Node)
-
process_first_pass
(node: dlmed.utils.json_scanner.Node)
-
process_second_pass
(node: dlmed.utils.json_scanner.Node)
-
start_config
(config_ctx: dlmed.utils.wfconf.ConfigContext)
-
-
evaluate_mmar
(mmar_root: str, wf_config_file_name=None, env_config_file_name=None, log_config_file_name=None, kv_list=None)
-
train_mmar
(mmar_root: str, wf_config_file_name=None, env_config_file_name=None, log_config_file_name=None, kv_list=None)