mindformers.trainer.ContrastiveLanguageImagePretrainTrainer

class mindformers.trainer.ContrastiveLanguageImagePretrainTrainer(model_name: str = None)[源代码]

Contrastive Language Image Pretrain Trainer.

参数

model_name (str) – The model name of Task-Trainer. Default: None

引发

NotImplementedError – If train method or evaluate method or predict method not implemented.

实际案例

>>> from mindformers import ContrastiveLanguageImagePretrainTrainer
>>> trainer = ContrastiveLanguageImagePretrainTrainer(model_name="clip_vit_b_b32")
>>> trainer.train()
train(config: Union[dict, mindformers.tools.register.config.MindFormerConfig, mindformers.trainer.config_args.ConfigArguments, mindformers.trainer.training_args.TrainingArguments, None] = None, network: Union[mindspore.nn.cell.Cell, mindformers.models.base_model.BaseModel, None] = None, dataset: Union[mindformers.dataset.base_dataset.BaseDataset, mindspore.dataset.engine.datasets_user_defined.GeneratorDataset, None] = None, optimizer: Optional[mindspore.nn.optim.optimizer.Optimizer] = None, wrapper: Optional[mindspore.nn.wrap.cell_wrapper.TrainOneStepCell] = None, callbacks: Union[mindspore.train.callback._callback.Callback, List[mindspore.train.callback._callback.Callback], None] = None, **kwargs)[源代码]

Train For Trainer.

参数

  • config (Optional[Union[dict, MindFormerConfig, ConfigArguments, TrainingArguments]]) – The task config which is used to configure the dataset, the hyper-parameter, optimizer, etc. It supports config dict or MindFormerConfig or TrainingArguments or ConfigArguments class. Default: None.

  • network (Optional[Union[Cell, BaseModel]]) – The network for trainer. It supports model name or BaseModel or MindSpore Cell class. Default: None.

  • dataset (Optional[Union[BaseDataset, GeneratorDataset]]) – The training dataset. It supports real dataset path or BaseDateset class or MindSpore Dataset class. Default: None.

  • optimizer (Optional[Optimizer]) – The optimizer used for training .Default: None.

  • wrapper (Optional[TrainOneStepCell]) – Wraps the network with the optimizer. It support TrainOneStepCell class of MindSpore. Default: None.

  • callbacks (Optional[Union[Callback, List[Callback]]]) – The training callback function. It supports CallBack or CallBack List of MindSpore. Default: None.