From 16f51f971607efab2c73958d207c582b389406c8 Mon Sep 17 00:00:00 2001 From: Leo gautheron Date: Thu, 20 Apr 2017 12:12:15 +0200 Subject: sinkhorn GPU implementation --- ot/gpu/cudamat/examples/util.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 ot/gpu/cudamat/examples/util.py (limited to 'ot/gpu/cudamat/examples/util.py') 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() + -- cgit v1.2.3