Skip to main content

classy.scripts.model.export

Functions​

export​

def export(
    model_name: str,
    no_strip: bool,
    all_ckpts: bool,
    zip_name: Optional[str] = None,
)
##

strip_checkpoint​

def strip_checkpoint(
    checkpoint_path: pathlib.Path,
    destination: pathlib.Path,
    keys_to_remove=('callbacks', 'optimizer_states', 'lr_schedulers'),
)
##

zip_run​

def zip_run(
    run: Run,
    tmpdir: pathlib.Path,
    zip_name: str = 'model.zip',
    strip_ckpt: bool = True,
    is_export: bool = False,
    best_only: bool = True,
) ‑> pathlib.Path
##