Skip to content

Commit

Permalink
Feature/add pytorch complex demo (#171)
Browse files Browse the repository at this point in the history
  • Loading branch information
Superjomn authored and deqingli committed Jan 17, 2018
1 parent b15aa0f commit 10f1dd3
Show file tree
Hide file tree
Showing 4 changed files with 33 additions and 2 deletions.
3 changes: 3 additions & 0 deletions .gitmodules
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
[submodule "demo/pytorch-CycleGAN-and-pix2pix"]
path = demo/pytorch-CycleGAN-and-pix2pix
url = https://github.com/Superjomn/pytorch-CycleGAN-and-pix2pix.git
27 changes: 27 additions & 0 deletions demo/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# VisualDL demos

VisualDL supports Python and C++ based DL frameworks,
there are several demos for different platforms.

## PaddlePaddle
Locates in `./paddle`.

This is a visualization for `resnet` on `cifar10` dataset, we visualize the CONV parameters,
and there are some interesting patterns.

## PyTorch GAN
Locates in `./pytorch-CycleGAN-and-pix2pix`.

This submodule is forked from [pytorch-CycleGAN-and-pix2pix](
https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix),
great model and the generated fake images are really funny.

This demo only works with CycleGAN mode, read [CycleGAN train doc](https://github.com/Superjomn/pytorch-CycleGAN-and-pix2pix#cyclegan-traintest) and [changes to the original code](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/compare/master...Superjomn:master) for more information.

## MxNet Mnist
Locates in `./mxnet_demo`.

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.
1 change: 1 addition & 0 deletions demo/pytorch-CycleGAN-and-pix2pix
4 changes: 2 additions & 2 deletions tests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -79,8 +79,8 @@ bigfile_reject() {
cd $TOP_DIR
# it failed to exclude .git, remove it first.
rm -rf .git
local largest_file="$(find . -path .git -prune -o -printf '%s %p\n' | sort -nr | head -n1)"
local size=$(echo $largest_file | awk '{print $1}')
local largest_file=$(find . -path .git -prune -o -printf '%s %p\n' | sort -nr | grep -v "CycleGAN"| head -n1)
local size=$(echo "$largest_file" | awk '{print $1}')
if [ "$size" -ge "$max_file_size" ]; then
echo $largest_file
echo "file size exceed $max_file_size"
Expand Down

0 comments on commit 10f1dd3

Please sign in to comment.