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

refine cv_image_defrcn trainer to avoid failed

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11622570
This commit is contained in:
shimin.ysm
2023-02-10 07:10:59 +00:00
committed by wenmeng.zwm
parent 6c5eb3bab3
commit 9b0e302a66
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/trainers/test_image_defrcn_fewshot_trainer.py
View File

@@ -66,18 +66,15 @@ class TestImageDefrcnFewShotTrainer(unittest.TestCase):
]
cfg.datasets.test = ['voc_2007_test_base{}'.format(split)]
cfg.input.min_size_test = 50
cfg.train.dataloader.ims_per_batch = 2
cfg.train.max_iter = 500
cfg.train.optimizer.lr = 0.01
cfg.train.dataloader.ims_per_batch = 4
cfg.train.max_iter = 300
cfg.train.optimizer.lr = 0.001
cfg.train.lr_scheduler.warmup_iters = 100
cfg.test.pcb_enable = False
return cfg
kwargs = dict(
model=self.model_id,
model_revision='v1.3.0',
cfg_modify_fn=base_cfg_modify_fn)
kwargs = dict(model=self.model_id, cfg_modify_fn=base_cfg_modify_fn)
trainer = build_trainer(
name=Trainers.image_fewshot_detection, default_args=kwargs)
trainer.train()