Skip to content

CLI command for inspecting checkpoints #5537

@wyli

Description

@wyli

Is your feature request related to a problem? Please describe.

python -m monai.bundle inspect_ckpt

to display some information about the variables and shapes saved. I saw similar util in tensorflow:

https://github.com/tensorflow/tensorflow/blob/064c7a3120f1571d7c90a777c787571787973518/tensorflow/python/tools/inspect_checkpoint.py#L163-L171

would also be good to have an option of computing hash value, it's used when creating a large_files.yml such as https://github.com/Project-MONAI/model-zoo/blob/dev/models/spleen_ct_segmentation/large_files.yml

cc @yiheng-wang-nv @Nic-Ma

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions