lyhue1991/torchkeras
A unified PyTorch training template library that provides a clean, ~200-line Keras-style training loop with visualization and callbacks.

torchkeras is a general-purpose PyTorch model training template designed to standardize the training workflow across different model libraries. It abstracts away the complexity of writing custom training loops by providing a clean, modular interface with built-in support for progress bars, evaluation metrics, early-stopping, and visualization. The tool includes callbacks for TensorBoard and Weights & Biases integration, aiming to replace the inconsistent training implementations found across various PyTorch ecosystems.