From ce0480f7ed780901d4339eaedfd87fc8a2a4ec2d Mon Sep 17 00:00:00 2001 From: "baiguan.yt" Date: Mon, 5 Dec 2022 11:43:52 +0800 Subject: [PATCH] update image-portait-enhancement trainer Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10717891 --- tests/trainers/test_image_portrait_enhancement_trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/trainers/test_image_portrait_enhancement_trainer.py b/tests/trainers/test_image_portrait_enhancement_trainer.py index 123e0098..a9fc74cb 100644 --- a/tests/trainers/test_image_portrait_enhancement_trainer.py +++ b/tests/trainers/test_image_portrait_enhancement_trainer.py @@ -61,6 +61,7 @@ class TestImagePortraitEnhancementTrainer(unittest.TestCase): train_dataset=self.dataset_train, eval_dataset=self.dataset_val, device='gpu', + max_epochs=1, work_dir=self.tmp_dir) trainer = build_trainer( @@ -81,7 +82,7 @@ class TestImagePortraitEnhancementTrainer(unittest.TestCase): train_dataset=self.dataset_train, eval_dataset=self.dataset_val, device='gpu', - max_epochs=2, + max_epochs=1, work_dir=self.tmp_dir) trainer = build_trainer(