dcslin edited a comment on issue #691:
URL: https://github.com/apache/singa/issues/691#issuecomment-629634540
Quoted from @joddiy , `m=MyModel(path) # check the file is a model or just
states`, this case, user know `MyModel` class, so just loading states right?
From the perspective of a new onnx user, please let me know if this part is
not correct.
`singa_rep`: model like `singa.Module`
`singa.save()/singa.load()`: save/load model w/o states
`singa.Module.save_states()/load_states()`: save/load model states only
### Use case 1, load model from onnx file
```python
class MySONNXModel(SONNXModel):
pass
# load from onnx model
onnx_model=onnx.load('./saved_models/onnx_model_downloaded')
m1=MySONNXModel(onnx_model)
m1.compile([placeholder_x], ...)
for _ in data:
m1.train_one_batch(_)
```
### use case 2: save states and model
```python
# save
m1.save_states('./saved_models/my_checkpoint_1')
singa.save('./saved_models/my_model_1')
```
### use case 3 load model and states from disk
```python
# Later reuse the model
m2=singa.load('./saved_models/my_model_1')
m2.load_states('./saved_models/my_checkpoint_1')
m2.compile([placeholder_x], use_graph=True)
```
### use case 4 load states only
```python
# model is known
class MyModel(Module):
pass
m3=MyModel(states_path='./saved_models/my_checkpoint_1') # could only be
states, right?
# m3=MyModel('./saved_models/my_model_1') # could not be saved_model right?
m3.compile(...)
```
To be frank, I am a bit overwhelmed by all the discussions not just in this
issue, is it possible to consolidate the new API into a specification including
example in singa-doc? Which is useful for new users? btw, is API in
[onnx-doc](https://github.com/apache/singa-doc/blob/master/docs-site/docs/onnx.md)
gonna change?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
