Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 20:19:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

update image-portait-enhancement trainer

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10717891
This commit is contained in:
baiguan.yt
2022-12-05 11:43:52 +08:00
committed by wenmeng.zwm
parent 941dbe75cf
commit ce0480f7ed
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/trainers/test_image_portrait_enhancement_trainer.py
View File

@@ -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(