Metadata-Version: 2.3
Name: ai-common-utilss
Version: 0.1.1
Summary: 
Author: venkatkedar007
Author-email: abc@gmail.com
Requires-Python: >=3.12
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Requires-Dist: ai-text-utils
Requires-Dist: numpy
Requires-Dist: torch
Description-Content-Type: text/markdown

# AI Common Utils - Trainer Module

This repository contains a reusable trainer class for PyTorch model training.

## Installation

```bash
pip install ai-common-utils
```

## Usage:

```python

from ai_common_utils.trainer import Trainer

trainer = Trainer()
train_losses, train_accuracies, val_losses, val_accuracies= trainer.train(model=model,
                                                                        train_dl=train_dl,
                                                                        val_dl=val_dl,
                                                                        epochs=epochs,
                                                                        loss_fn=loss_fn,
                                                                        optimizer=optimizer,
                                                                        status_freq=20,
                                                                        checkpoint_freq=40,
                                                                        chk_point_file='model_weights.pth',
                                                                        device=device
                                                                        )
```

## Details
- **Returns**: Lists of (train_losses, train_accuracies, val_losses, val_accuracies) per epoch  
- **Progress**: Prints train/val metrics every `status_freq` batches  
- **Checkpoints**: Saves model weights to `chk_point_file` every `checkpoint_freq` batches

