From 68754205df8e7edcf4f65f5bca6e8b69eceba23d Mon Sep 17 00:00:00 2001 From: daminglu Date: Wed, 31 Jan 2018 18:26:14 +0800 Subject: [PATCH] Add readme pytorch (#233) --- demo/README.md | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/demo/README.md b/demo/README.md index 4720b53b..334f5a92 100644 --- a/demo/README.md +++ b/demo/README.md @@ -25,3 +25,10 @@ By adding VisualDL as callbacks to `model.fit`, we can use the Python SDK in MxNet, but it seems that only the outside program can only retrieve parameters in epoch callbacks, 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. -- GitLab