mmpretrain/docs/en/user_guides/useful_tools.md

1.7 KiB

Useful Tools (TODO)

Print the entire config

tools/misc/print_config.py prints the whole config verbatim, expanding all its imports.

python tools/misc/print_config.py ${CONFIG} [--cfg-options ${CFG_OPTIONS}]

Description of all arguments:

  • config : The path of the model config file.
  • --cfg-options: If specified, the key-value pair config will be merged into the config file, for more details please refer to Learn about Configs

Examples:

python tools/misc/print_config.py configs/t2t_vit/t2t-vit-t-14_8xb64_in1k.py

Verify Dataset

tools/misc/verify_dataset.py can verify dataset, check whether there are broken pictures in the given dataset.

python tools/print_config.py \
    ${CONFIG} \
    [--out-path ${OUT-PATH}] \
    [--phase ${PHASE}] \
    [--num-process ${NUM-PROCESS}]
    [--cfg-options ${CFG_OPTIONS}]

Description of all arguments:

  • config : The path of the model config file.
  • --out-path : The path to save the verification result, if not set, defaults to 'brokenfiles.log'.
  • --phase : Phase of dataset to verify, accept "train" "test" and "val", if not set, defaults to "train".
  • --num-process : number of process to use, if not set, defaults to 1.
  • --cfg-options: If specified, the key-value pair config will be merged into the config file, for more details please refer to Learn about Configs

Examples:

python tools/misc/verify_dataset.py configs/t2t_vit/t2t-vit-t-14_8xb64_in1k.py --out-path broken_imgs.log --phase val --num-process 8

FAQs

  • None