fix #967

scripts/checkpoint-manipulate.py

@@ -4,7 +4,7 @@
 
 
 import numpy as np
-from tensorpack.tfutils.varmanip import dump_chkpt_vars
+from tensorpack.tfutils.varmanip import load_chkpt_vars
 from tensorpack.utils import logger
 import argparse
 
@@ -21,13 +21,13 @@ if __name__ == '__main__':
     elif args.model.endswith('.npz'):
         params = dict(np.load(args.model))
     else:
-        params = dump_chkpt_vars(args.model)
+        params = load_chkpt_vars(args.model)
     logger.info("Variables in the model:")
     logger.info(str(params.keys()))
 
     if args.dump:
         assert args.dump.endswith('.npz'), args.dump
-        np.save(args.dump, **params)
+        np.savez(args.dump, **params)
 
     if args.shell:
         # params is a dict. play with it