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',
'image').rename_column('answer:Value', 'answer'))

self.max_epochs = 3
self.max_epochs = 2

def tearDown(self):
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.train.max_epochs = 3
cfg.train.max_epochs = 2
cfg.train.lr_scheduler = {
'type': 'LinearLR',
'start_factor': 1.0,


Loading…
Cancel
Save