mirror of
https://github.com/wassname/viz_torch_optim.git
synced 2026-06-27 16:32:39 +08:00
8efe458f40bd1de5e68e850d99aa83ccf62ea197
This project generates animations of pytorch optimizers solving toy problems. Examples Below.
Some nice animations were posted a few years ago by Alex Radford but didn't include the Adam optimizer or landscapes with noise. Louis Tiao blogged about how to make the visualizations. The pytorch unit tests show how to run the optimizers on test functions. I pulled these together and shared the result at https://github.com/wassname/viz_torch_optim. Please make some better animations and share them.
Examples
Usage:
git clone https://github.com/wassname/viz_torch_optimjupyter notebook- open main.ipynb
- install any missing dependencies with pip
Description
Languages
Jupyter Notebook
100%







