mindformers.trainer

MindFormers Trainer API.

mindformers.trainer.Trainer

Executor of general task trainers.

mindformers.trainer.BaseTrainer

Base Task Trainer. Args: task (str): The task name supported. model_name (str): The model name of Task-Trainer. Default: None Raises: NotImplementedError: If train method or evaluate method or predict method not implemented.

mindformers.trainer.TrainingArguments

TrainingArguments is the subset of the arguments we use in our default config which is related to the training in MindSpore.

mindformers.trainer.ContextConfig

Context Config For Running Environment.

mindformers.trainer.ParallelContextConfig

Parallel Context Config.

mindformers.trainer.TranslationTrainer

Trainer of translation task.

mindformers.trainer.CausalLanguageModelingTrainer

CausalLanguageModelingTrainer Task For Trainer.

mindformers.trainer.ContrastiveLanguageImagePretrainTrainer

Contrastive Language Image Pretrain Trainer.

mindformers.trainer.ImageToTextGenerationTrainer

ImageToTextGenerationTrainer Task For Trainer.

mindformers.trainer.ImageClassificationTrainer

ImageClassification Task For Trainer.

mindformers.trainer.MaskedImageModelingTrainer

MaskedImageModeling Task For Trainer.

mindformers.trainer.MaskedLanguageModelingTrainer

MaskedLanguageModeling Task For Trainer.

mindformers.trainer.QuestionAnsweringTrainer

Trainer of question answering task.

mindformers.trainer.TextClassificationTrainer

Trainer of text classification task.

mindformers.trainer.ZeroShotImageClassificationTrainer

Trainer of zero-shot image classification task.