Last released Dec 2, 2024
A lightweight Python package for PyTorch that enhances readability by color-coding module prints based on their trainable status.
Supported by