Trainer

TrainerError ¶

Base errors for all rag trainer relevant exceptions.

Source code in src/fed_rag/exceptions/trainer.py

class TrainerError(FedRAGError):
    """Base errors for all rag trainer relevant exceptions."""

    pass

Raised if underlying datasets between dataloaders are inconsistent.

Source code in src/fed_rag/exceptions/trainer.py

class InconsistentDatasetError(TrainerError):
    """Raised if underlying datasets between dataloaders are inconsistent."""

    pass

Raised if an unexpected loss is attached to a trainer object.

Source code in src/fed_rag/exceptions/trainer.py

class InvalidLossError(TrainerError):
    """Raised if an unexpected loss is attached to a trainer object."""

    pass

Raised if an invalid data collator is attached to a trainer object.

Source code in src/fed_rag/exceptions/trainer.py

class InvalidDataCollatorError(TrainerError):
    """Raised if an invalid data collator is attached to a trainer object."""

    pass

Raised if a required tensor has not been supplied in the inputs.

Source code in src/fed_rag/exceptions/trainer.py

class MissingInputTensor(TrainerError):
    """Raised if a required tensor has not been supplied in the inputs."""

    pass