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:
[email protected]
With regards,
Apache Git Services
