last_checkpoint.npz test.py