Skip to content

Commit 34a2008

Browse files
authored
Merge pull request #35 from Yoctol/update-readme
Update README.md
2 parents 5038b9f + 3453268 commit 34a2008

2 files changed

Lines changed: 17 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

serving_utils/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
from .client import Client, PredictInput
22
from .saver import Saver
3+
from .loader import Loader

0 commit comments

Comments
 (0)