diff options
Diffstat (limited to 'ot/gpu/cudamat/examples/util.py')
-rw-r--r-- | ot/gpu/cudamat/examples/util.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/ot/gpu/cudamat/examples/util.py b/ot/gpu/cudamat/examples/util.py new file mode 100644 index 0000000..79ceead --- /dev/null +++ b/ot/gpu/cudamat/examples/util.py @@ -0,0 +1,22 @@ +from __future__ import division +import gzip +try: import cPickle as pickle +except: import pickle + +def save(fname, var_list, source_dict): + var_list = [var.strip() for var in var_list.split() if len(var.strip())>0] + fo = gzip.GzipFile(fname, 'wb') + pickle.dump(var_list, fo) + for var in var_list: + pickle.dump(source_dict[var], fo, protocol=2) + fo.close() + +def load(fname, target_dict, verbose = True): + fo = gzip.GzipFile(fname, 'rb') + var_list = pickle.load(fo) + if verbose: + print(var_list) + for var in var_list: + target_dict[var] = pickle.load(fo) + fo.close() + |