From 9952866118197dd19ff39a26224fc93a70cbd14e Mon Sep 17 00:00:00 2001 From: "Yilin (Eleen) Bao" <62961093+yilin-bao@users.noreply.github.com> Date: Sat, 9 Dec 2023 02:45:15 -0600 Subject: [PATCH] Update README.md --- README.md | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/README.md b/README.md index 8b3c0ee..be4a21a 100644 --- a/README.md +++ b/README.md @@ -3,8 +3,6 @@ [![Build Status](https://travis-ci.org/user/repo.svg?branch=master)](https://travis-ci.org/user/repo) [![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT) -![Alt text](Attention.drawio.png) - ## Overview This library is designed to generate visualizations of neural networks by analyzing PyTorch code. Unlike conventional methods, our approach eliminates the need for large parameter files (pth) or specific input formats. With minimal AI knowledge, users can leverage this tool to create visualizations that aid in understanding complex networks, including support for transformers and attention mechanisms. @@ -120,23 +118,25 @@ This project is licensed under the [MIT License](https://opensource.org/licenses ## Project Status -Active development status or any relevant project status updates. - ## Community and Contact -Engage with the community on our [forum](https://community-forum-link.com) or contact us via email at [your-email@example.com]. - -## Badges - -Display badges indicating project status, build, version, and other relevant information. +Contact us via email at [yibao@ucsd.edu]. ## Screenshots or Demos -Include screenshots or links to demos showcasing the visualizations generated by the library. +![Alt text](Attention.drawio.png) ## Related Projects -Explore other projects or tools related to neural network visualization. +- https://github.com/gwding/draw_convnet +- http://alexlenail.me/NN-SVG/LeNet.html +- https://github.com/HarisIqbal88/PlotNeuralNet +- https://www.tensorflow.org/tensorboard/graphs +- https://github.com/BVLC/caffe/blob/master/python/caffe/draw.py +- https://www.mathworks.com/help/nnet/ref/view.html +- https://transcranial.github.io/keras-js/%23/inception-v3 +- https://github.com/stared/keras-sequential-ascii/ +- https://github.com/lutzroeder/Netron ## Frequently Asked Questions (FAQ)