-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Description
Is your feature request related to a problem? Please describe.
Given a bundle with saved weights in the models directory and a network that's compatible with Torchscript, the command line to use ckpt_export gets rather long:
python -m monai.bundle ckpt_export --net_id network_def --bundle_root . --meta_file configs/metadata.json --config_file configs/inference.json --filepath models/model.ts --ckpt_file models/model.pt --key_in_ckpt modelWould it be possible to simplify this by providing defaults for meta_file and others, or try to guess, if the loaded checkpoint is a dictionary, what key_in_ckpt would be?
Describe the solution you'd like
If the command line would be:
python -m monai.bundle ckpt_export --bundle_root . --config_file configs/inference.json --filepath models/model.tswith everything omitted from the previous version replaced with defaults or guesses. If key_in_ckpt is needed that's fine but trying to guess and saying in the output what was guessed would often work. Similarly a defaults of configs/metadata.json for the meta file would usually work.
Describe alternatives you've considered
Keep using the old command line, it does work so this is about usability.
Additional context
The commands need more tutorial materials which is a work in progress.