Browse Source

update image-portait-enhancement trainer

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10717891
master^2
baiguan.yt wenmeng.zwm 3 years ago
parent
commit
ce0480f7ed
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      tests/trainers/test_image_portrait_enhancement_trainer.py

+ 2
- 1
tests/trainers/test_image_portrait_enhancement_trainer.py View File

@@ -61,6 +61,7 @@ class TestImagePortraitEnhancementTrainer(unittest.TestCase):
train_dataset=self.dataset_train, train_dataset=self.dataset_train,
eval_dataset=self.dataset_val, eval_dataset=self.dataset_val,
device='gpu', device='gpu',
max_epochs=1,
work_dir=self.tmp_dir) work_dir=self.tmp_dir)


trainer = build_trainer( trainer = build_trainer(
@@ -81,7 +82,7 @@ class TestImagePortraitEnhancementTrainer(unittest.TestCase):
train_dataset=self.dataset_train, train_dataset=self.dataset_train,
eval_dataset=self.dataset_val, eval_dataset=self.dataset_val,
device='gpu', device='gpu',
max_epochs=2,
max_epochs=1,
work_dir=self.tmp_dir) work_dir=self.tmp_dir)


trainer = build_trainer( trainer = build_trainer(


Loading…
Cancel
Save