Skip to content

Commit 3453268

Browse files
committed
update readme for loader
1 parent bbce218 commit 3453268

1 file changed

Lines changed: 16 additions & 2 deletions

File tree

README.md

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,12 @@ From Github repository:
3030

3131
## Usage
3232

33-
1. Saver
33+
1. Saver and Loader
3434
```python
3535
import tensorflow as tf
3636

37-
from serving_utils import Saver
37+
from serving_utils.saver import Saver
38+
from serving_utils.loader import Loader
3839

3940
saver = Saver(
4041
session=tf.Session(graph=your_graph),
@@ -48,6 +49,14 @@ saver = Saver(
4849
freeze=True, # (default: True) Frozen graph will be saved if True.
4950
)
5051
saver.save()
52+
53+
loader = Loader(
54+
path='/path/to/serving',
55+
# version=1, # if not specified, use the latest version
56+
)
57+
new_sess = tf.Session()
58+
59+
loader.load(new_sess) # load the saved model into new session
5160
```
5261

5362
2. Client
@@ -73,6 +82,11 @@ frozen_graph_def = freeze_graph(session, output_op_names)
7382
new_session = create_session_from_graphdef(frozen_graph_def)
7483
```
7584

85+
4. Loader
86+
```python
87+
88+
```
89+
7690

7791
## Test
7892

0 commit comments

Comments
 (0)