jayparmr's picture
Update infer/lib/train/process_ckpt.py
3f3296a