jayparmr's picture
Update infer/lib/train/process_ckpt.py
4a2b5fd