--- a +++ b/utils/file_utils.py @@ -0,0 +1,13 @@ +import pickle +import h5py + +def save_pkl(filename, save_object): + writer = open(filename,'wb') + pickle.dump(save_object, writer) + writer.close() + +def load_pkl(filename): + loader = open(filename,'rb') + file = pickle.load(loader) + loader.close() + return file