{"id":19820290,"url":"https://github.com/strcoder4007/training-visualization","last_synced_at":"2026-05-15T03:33:19.289Z","repository":{"id":249367593,"uuid":"831318475","full_name":"strcoder4007/Training-Visualization","owner":"strcoder4007","description":"A tool to visualize the hidden layers of the model in real-time.","archived":false,"fork":false,"pushed_at":"2024-07-22T07:00:07.000Z","size":22954,"stargazers_count":0,"open_issues_count":0,"forks_count":0,"subscribers_count":1,"default_branch":"main","last_synced_at":"2025-02-28T20:22:55.053Z","etag":null,"topics":["fashion-mnist","layer-visualization","pytorch"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/strcoder4007.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2024-07-20T07:53:51.000Z","updated_at":"2024-07-22T07:00:10.000Z","dependencies_parsed_at":"2025-01-11T08:36:57.954Z","dependency_job_id":null,"html_url":"https://github.com/strcoder4007/Training-Visualization","commit_stats":null,"previous_names":["strcoder4007/training-visualization"],"tags_count":0,"template":false,"template_full_name":null,"purl":"pkg:github/strcoder4007/Training-Visualization","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/strcoder4007%2FTraining-Visualization","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/strcoder4007%2FTraining-Visualization/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/strcoder4007%2FTraining-Visualization/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/strcoder4007%2FTraining-Visualization/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/strcoder4007","download_url":"https://codeload.github.com/strcoder4007/Training-Visualization/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/strcoder4007%2FTraining-Visualization/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":33052702,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-05-13T13:14:54.681Z","status":"online","status_checked_at":"2026-05-15T02:00:06.351Z","response_time":103,"last_error":null,"robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":true,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["fashion-mnist","layer-visualization","pytorch"],"created_at":"2024-11-12T10:22:40.714Z","updated_at":"2026-05-15T03:33:19.274Z","avatar_url":"https://github.com/strcoder4007.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Hidden Layer Visualization while Training\nA tool to visualize the hidden layers of the model in real-time. \nIn this project, we delve in a fundamental yet often overlooked aspect - visualizing the layers of the neural network.\n\nBy visually inspecting these weights during training or inference, we can gain insights into overfitting, underfitting, convergence issues, and even help in designing better architectures. \nVisualizing the weights of neural network layers can reveal patterns, identify bottlenecks, and guide us towards more efficient models.\n\n\n## Visualization of all the weights\n![Weight Visualization](/images/weights.gif)\n\n## Visualization of all the layers and their activations\n![Layer Visualization](/images/layers.gif)\n\n## How to use this repo\n1. Clone the repository:\n   ```bash\n   git clone https://github.com/strcoder4007/Training-Visualization.git\n   cd Training-Visualization\n   ```\n\n2. Install the required dependencies:\n   ```bash\n   pip install -r requirements.txt\n   ```\n   This command will install all necessary packages including `Pytorch`, `opencv-python`, `imagio`, and `pickle`.\n\n3. Dataset is present in the root of this repo, to run the model, run `main.py`:\n   ```bash\n   python main.py\n   ```\n   The `.pkl` data file with all the weights and biases of the model for every epoch and step will be generated and saved in the root of the folder\n\n4. `main.py` will use the `.pkl` file to load all the weights and biases to display them using `Tkinter`","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fstrcoder4007%2Ftraining-visualization","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fstrcoder4007%2Ftraining-visualization","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fstrcoder4007%2Ftraining-visualization/lists"}