View source on GitHub
|
Create trainer.
tfm.core.train_utils.create_trainer(
params: tfm.core.base_trainer.ExperimentConfig,
task: tfm.core.base_task.Task,
train: bool,
evaluate: bool,
checkpoint_exporter: Optional[tfm.core.train_utils.BestCheckpointExporter] = None,
trainer_cls=tfm.core.base_trainer.Trainer
) -> tfm.core.base_trainer.Trainer
View source on GitHub