Introducing the Loss Landscape Animation Python Package

In my previous article Visualizing Optimization Trajectory of Neural Nets, I showed a number of animated figures to demonstrate the training process of neural networks. Since then some readers expressed interest in the code, so I ported it to a Python package and published it on PyPI

In my previous article Visualizing Optimization Trajectory of Neural Nets, I showed a number of animated figures to demonstrate the training process of neural networks. Since then some readers expressed interest in the code, so I ported it to a Python package and published it on PyPI. With this package, you can produce similar plots easily, experiment with the default datasets and default models out-of-the-box, or plot for your own data modules and/or models by implementing an interface in the style of PyTorch Lightning.

Installation

To install directly from PyPI

pip install loss-landscape-anim

Or you can install from source directly. I recommend using Poetry. If you don’t know what Poetry is, I highly recommend you check it out. It’s a really neat dependency manager that will make your life easier. Once you have cloned this repo and installed Poetry, just run the command below in the project directory to install the dependencies.

poetry shell
poetry install

Without Poetry, pip install -r requirements.txt should also work.

machine-learning python deep-learning

PyTorch for Deep Learning | Data Science | Machine Learning | Python. PyTorch is a library in Python which provides tools to build deep learning models. What python does for programming PyTorch does for deep learning. Python is a very flexible language for programming and just like python, the PyTorch library provides flexible tools for deep learning.

In this video, Deep Learning Tutorial with Python | Machine Learning with Neural Networks Explained, Frank Kane helps de-mystify the world of deep learning and artificial neural networks with Python!

We supply you with world class machine learning experts / ML Developers with years of domain experience who can add more value to your business.

PyTorch is a library in Python which provides tools to build deep learning models. What python does for programming PyTorch does for deep learning.

We supply you with world class machine learning experts / ML Developers with years of domain experience who can add more value to your business.