medl.tools.mmar_gen package

class MMARGenerator(template_json_path)

Bases: object

create_mmar(params, dest_path)

Create mmar from given paramters

Parameters
  • params (dict) – parameter dictionary.

  • dest_path (str) – Path to place output mmar.

load_template(template_json_path)

Load templates from json configuration.

Parameters

template_json_path (str) – Path to json configuration.

class Template(default_dir)

Bases: object

Template for mmar generator

Parameters

default_dir (str) – Path to default directory for this template.

add_template_item(category, item_dict)
get_category_option(category, option_name)
get_default_dir()
class TemplateCategory

Bases: object

AUTOML = 'automl'
DEFAULT = 'default'
FL = 'fl'
TASK = 'task'
get_categories()
valid_category()
validate_template_item_dict(item_dict)
© Copyright 2021, NVIDIA. Last updated on Feb 2, 2023.