Neural Network Training Progress Visualization

Visualizing tool for PyTorch NN models.

Requirements

The tool requires following dependencies to run:

Python>=3.8
pip>=20

If CUDA will be used:

CUDA>=11.1

Quickstart

Install with pip install nnvis