Browse Source
reduce image denoising test time
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10712244master
huizheng.hz
yingda.chen
3 years ago
1 changed files with 3 additions and 3 deletions
Unified View
Diff Options
+ 3
- 3
tests/trainers/test_image_denoise_trainer.py
View File
| @@ -62,7 +62,7 @@ class ImageDenoiseTrainerTest(unittest.TestCase): | |||||
| trainer.train() | trainer.train() | ||||
| results_files = os.listdir(self.tmp_dir) | results_files = os.listdir(self.tmp_dir) | ||||
| self.assertIn(f'{trainer.timestamp}.log.json', results_files) | self.assertIn(f'{trainer.timestamp}.log.json', results_files) | ||||
| for i in range(2): | |||||
| for i in range(1): | |||||
| self.assertIn(f'epoch_{i+1}.pth', results_files) | self.assertIn(f'epoch_{i+1}.pth', results_files) | ||||
| @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') | @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') | ||||
| @@ -73,13 +73,13 @@ class ImageDenoiseTrainerTest(unittest.TestCase): | |||||
| model=model, | model=model, | ||||
| train_dataset=self.dataset_train, | train_dataset=self.dataset_train, | ||||
| eval_dataset=self.dataset_val, | eval_dataset=self.dataset_val, | ||||
| max_epochs=2, | |||||
| max_epochs=1, | |||||
| work_dir=self.tmp_dir) | work_dir=self.tmp_dir) | ||||
| trainer = build_trainer(default_args=kwargs) | trainer = build_trainer(default_args=kwargs) | ||||
| trainer.train() | trainer.train() | ||||
| results_files = os.listdir(self.tmp_dir) | results_files = os.listdir(self.tmp_dir) | ||||
| self.assertIn(f'{trainer.timestamp}.log.json', results_files) | self.assertIn(f'{trainer.timestamp}.log.json', results_files) | ||||
| for i in range(2): | |||||
| for i in range(1): | |||||
| self.assertIn(f'epoch_{i+1}.pth', results_files) | self.assertIn(f'epoch_{i+1}.pth', results_files) | ||||