From ccff185281d8b5475b7fd5642d6f271cd29799b2 Mon Sep 17 00:00:00 2001 From: Willy Fitra Hendria Date: Mon, 19 Feb 2024 05:26:01 +0900 Subject: [PATCH] Update version (#5) * Update README.md * Update setup.py * Update image * updated by prettier --- README.md | 4 ++-- setup.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 061333b..0cac4dd 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,7 @@ **VisualTorch** aims to help visualize Torch-based neural network architectures. Currently, this package supports generating layered-style architectures for Torch Sequential models. This package is adapted from [visualkeras](https://github.com/paulgavrikov/visualkeras) by [@paulgavrikov](https://github.com/paulgavrikov). -**v0.0.1**: Support for layered architecture of torch Sequential. +**v0.1**: Support for layered architecture of torch Sequential. ## Installation @@ -47,7 +47,7 @@ visualtorch.layered_view(model, input_shape=input_shape, legend=True, to_file='o visualtorch.layered_view(model, input_shape=input_shape, legend=True, to_file='output.png').show() # write and show ``` -![simple-cnn](https://github.com/willyfh/visualtorch/assets/5786636/7cd10f94-0ddb-4b9d-a4e0-614e55f2ddee) +![simple-cnn](https://github.com/willyfh/visualtorch/assets/5786636/9b646fac-c336-4253-ac01-8f3e6b2fcc0b) ## Contributing diff --git a/setup.py b/setup.py index 6ea7e32..007f357 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ setuptools.setup( name="visualtorch", - version="0.0.1", + version="0.1", author="Willy Fitra Hendria", author_email="willyfitrahendria@gmail.com", description="Architecture visualization of Torch models",