未验证 提交 68754205 编写于 作者: D daminglu 提交者: GitHub

Add readme pytorch (#233)

上级 2ff30dd9
...@@ -25,3 +25,10 @@ By adding VisualDL as callbacks to `model.fit`, ...@@ -25,3 +25,10 @@ By adding VisualDL as callbacks to `model.fit`,
we can use the Python SDK in MxNet, we can use the Python SDK in MxNet,
but it seems that only the outside program can only retrieve parameters in epoch callbacks, but it seems that only the outside program can only retrieve parameters in epoch callbacks,
that limits the number of steps for visualization. that limits the number of steps for visualization.
## PyTorch CNN
Locates in `./pytorch`.
This shows how to use VisualDL in PyTorch for a CNN on `cifar10` dataset. We visualize the loss in Scalar,
two convolutional layers in Image, the change trend of conv1 weights in Histogram and the final model graph
in Graph.
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册