Overview Version History Q & A Rating & Review
ARC Lens
ARC Lens is a real-time training visualization dashboard and metrics tracker extension for PyTorch. It helps visualize training losses, learning rates, gradient norms, and GPU memory usage.
Features
Real-Time PyTorch Monitoring : Visualizes losses, learning rates, gradient norms, and GPU memory usage side-by-side.
Interactive Charts : Smooth interactive charts with mouse drag-to-pan, scroll-to-zoom, and auto-scroll pausing.
Environment Verification : Detects if training components are available in the active Python environment.
Getting Started
Open a Python (.py) training script in VS Code.
Click the Run with ARC Lens button in the top-right toolbar.
License
This extension is licensed under the MIT License.