def METHOD_NAME(self):
    kwargs = dict(
        model=self.model_id,
        train_dataset=self.dataset_train,
        eval_dataset=self.dataset_val,
        work_dir=self.tmp_dir)
    trainer = build_trainer(default_args=kwargs)
    trainer.train()
    results_files = os.listdir(self.tmp_dir)
    self.assertIn(f'{trainer.timestamp}.log.json', results_files)
    for i in range(1):
        self.assertIn(f'epoch_{i+1}.pth', results_files)
    pipeline_colorization = pipeline(
        task=Tasks.image_colorization, model=f'{self.tmp_dir}/output')
    pipeline_colorization('data/test/images/marilyn_monroe_4.jpg')