modlee.modlee_image_model module

class modlee.modlee_image_model.ImageCallback(num_classes=None, *args, **kwargs)

Bases: Callback

Saves accuracy

on_validation_batch_end(trainer: Trainer, pl_module: LightningModule, outputs: Tensor | Dict[str, Any] | None, batch: Any, batch_idx: int, dataloader_idx: int = 0) → None

Called when the validation batch ends.

class modlee.modlee_image_model.ModleeImageModel(task='classification', num_classes=None, *args, **kwargs)

Bases: ModleeModel

Subclass of ModleeModel with image-specific convenience wrappers - Logs classification accuracy - Calculates data-specific data statistics

configure_callbacks()

Configure callbacks for auto-documentation.

Returns:

A list of callbacks for auto-documentation.