Browse Source

[to #42322933] reduce train epoch from 3 to w

master
Yingda Chen 3 years ago
parent
commit
39a309b655
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      tests/trainers/test_finetune_mplug.py
  2. +1
    -1
      tests/trainers/test_finetune_token_classificatin.py

+ 1
- 1
tests/trainers/test_finetune_mplug.py View File

@@ -35,7 +35,7 @@ class TestFinetuneMPlug(unittest.TestCase):
}).rename_column('image:FILE', }).rename_column('image:FILE',
'image').rename_column('answer:Value', 'answer')) 'image').rename_column('answer:Value', 'answer'))


self.max_epochs = 3
self.max_epochs = 2


def tearDown(self): def tearDown(self):
shutil.rmtree(self.tmp_dir) shutil.rmtree(self.tmp_dir)


+ 1
- 1
tests/trainers/test_finetune_token_classificatin.py View File

@@ -92,7 +92,7 @@ class TestFinetuneTokenClassification(unittest.TestCase):
} }
} }
cfg['preprocessor'] = {'type': 'token-cls-tokenizer'} cfg['preprocessor'] = {'type': 'token-cls-tokenizer'}
cfg.train.max_epochs = 3
cfg.train.max_epochs = 2
cfg.train.lr_scheduler = { cfg.train.lr_scheduler = {
'type': 'LinearLR', 'type': 'LinearLR',
'start_factor': 1.0, 'start_factor': 1.0,


Loading…
Cancel
Save