[Pytorch] Make test script generate checkpoints if they don't exist#2650

ptrendx merged 1 commit intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

kainzhong:fix/generate_ckpt_for_testkainzhong/TransformerEngine:fix/generate_ckpt_for_testCopy head branch name to clipboard