singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [singa-doc] nudles commented on a change in pull request #9: Documentation about computational graph
Date Thu, 02 Apr 2020 14:18:51 GMT
nudles commented on a change in pull request #9: Documentation about computational graph
URL: https://github.com/apache/singa-doc/pull/9#discussion_r402343986
 
 

 ##########
 File path: docs-site/docs/autograd.md
 ##########
 @@ -189,3 +193,67 @@ for epoch in range(epochs):
         for p, gp in autograd.backward(loss):  # auto backward
             sgd.update(p, gp)
 ```
+
+### Operation + Module
+
+The following [example](https://github.com/apache/singa/blob/master/examples/autograd/cnn_module.py)
implements a CNN model using the Module provided by the module.
+
+#### Define the model class
+
+Define the model class, it should be the subclass of the Module. In this way, all operations
used during traing phase  will form a calculation graph and will be analyzed. The operations
in the graph will be scheduled and executed efficiently. Layers can also be included in the
module class.
 
 Review comment:
   subclass of Module

----------------------------------------------------------------
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


With regards,
Apache Git Services

Mime
View raw message