Metadata-Version: 2.1
Name: torch-training-loop
Version: 0.0.2
Summary: Simple Keras-like Training Loop for Pytorch.
Home-page: https://github.com/beekill95/torch-training-loop
License: MIT
Author: beekill
Author-email: nguyenmbquan95@gmail.com
Requires-Python: >=3.8,<4.0
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Dist: numpy
Requires-Dist: pandas
Requires-Dist: tensorboard
Requires-Dist: torch
Requires-Dist: torcheval
Requires-Dist: tqdm
Project-URL: Repository, https://github.com/beekill95/torch-training-loop
Description-Content-Type: text/markdown

[![Tests](https://github.com/beekill95/torch-training-loop/workflows/Tests/badge.svg)](https://github.com/beekill95/torch-training-loop/actions?query=workflow:"Tests")
[![License](https://img.shields.io/badge/License-MIT-blue)](#license)
[![PyPI - Version](https://img.shields.io/pypi/v/torch-training-loop)](https://pypi.org/project/torch-training-loop/)
![PyPI - Python Version](https://img.shields.io/pypi/pyversions/torch-training-loop)

⚠️The package is under development, expect bugs and breaking changes!

# Torch Training Loop

Simple Keras-like Training Loop for Pytorch.

## Installation

> pip install torch-training-loop

## Features

* Support Keras-like API for training Torch models;
* Support Keras-like callbacks for logging metrics to Tensorboard, model checkpoint,
and early stopping;
* Show training & validation progress via `tqdm`;
* Display metrics during training & validation via `torcheval`.

## License

Distributed under the MIT License. See `LICENSE.txt` for more information.

