Update for loading pretrained checkpoints

[gaozhihan]

Now users can run scripts for testing pretrained checkpoints without the need to specify the cfg.yaml files. The scripts load the correct cfgs for pretrained checkpoints automatically.

[sxjscience]

LGTM for now. We need to later borrow the design of timm and huggingface/transformers