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