a | b/utils/file_utils.py | ||
---|---|---|---|
1 | import pickle |
||
2 | import h5py |
||
3 | |||
4 | def save_pkl(filename, save_object): |
||
5 | writer = open(filename,'wb') |
||
6 | pickle.dump(save_object, writer) |
||
7 | writer.close() |
||
8 | |||
9 | def load_pkl(filename): |
||
10 | loader = open(filename,'rb') |
||
11 | file = pickle.load(loader) |
||
12 | loader.close() |
||
13 | return file |