singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [singa] dcslin edited a comment on issue #691: Add save and load method for Module class
Date Sat, 16 May 2020 12:01:03 GMT

dcslin edited a comment on issue #691:
URL: https://github.com/apache/singa/issues/691#issuecomment-629634540


   From the perspective of a new onnx user:
   
   ```python
   class MySONNXModel(SONNXModel):
       pass
   
   m1=MySONNXModel()
   m1.compile([placeholder], ...)
   for _ in data:
       m1.train_one_batch(_)
   
   m1.save_states('./saved_models/my_sonnx_model_onnx_format') # transferable to other deeplearning
framework
   singa.save('./saved_models/my_sonnx_model_singa_format') # only valid in singa
   
   
   # Later reuse the model
   onnx_model = onnx.load('./saved_models/my_sonnx_model_onnx_format')
   m2=MyONNXModel(onnx_model)
   m2.compile([x], use_graph=True)
   
   # m1 == m2?
   
   # load as singa model
   m3=singa.load('./saved_models/my_sonnx_model_singa_format')
   m3.compile([x], use_graph=True)
   
   # m1 == to_onnx(m3)?
   ```
   
   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:
users@infra.apache.org



Mime
View raw message