Browse Source

add five finetune task & merge master

master
翎航 2 years ago
parent
commit
3534970709
1 changed files with 0 additions and 7 deletions
  1. +0
    -7
      tests/trainers/test_ofa_trainer.py

+ 0
- 7
tests/trainers/test_ofa_trainer.py View File

@@ -71,13 +71,6 @@ class TestOfaTrainer(unittest.TestCase):

@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_trainer_std(self):
# WORKSPACE = './workspace/ckpts/recognition'
# os.makedirs(WORKSPACE, exist_ok=True)
#
# pretrained_model = 'damo/ofa_ocr-recognition_scene_base_zh'
# cfg = read_config(pretrained_model)
# config_file = os.path.join(WORKSPACE, ModelFile.CONFIGURATION)
# cfg.dump(config_file)
WORKSPACE = './workspace/ckpts/recognition'
os.makedirs(WORKSPACE, exist_ok=True)
config_file = os.path.join(WORKSPACE, ModelFile.CONFIGURATION)


Loading…
Cancel
Save