提交 97136fe9 编写于 作者: H Hui Zhang

trainer eval when not train

上级 fa5531c0
...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments ...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments
def main_sp(config, args): def main_sp(config, args):
exp = Tester(config, args) exp = Tester(config, args)
exp.setup() with exp.eval():
exp.run_export() exp.setup()
exp.run_export()
def main(config, args): def main(config, args):
......
...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments ...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments
def main_sp(config, args): def main_sp(config, args):
exp = Tester(config, args) exp = Tester(config, args)
exp.setup() with exp.eval():
exp.run_test() exp.setup()
exp.run_test()
def main(config, args): def main(config, args):
......
...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments ...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments
def main_sp(config, args): def main_sp(config, args):
exp = ExportTester(config, args) exp = ExportTester(config, args)
exp.setup() with exp.eval():
exp.run_test() exp.setup()
exp.run_test()
def main(config, args): def main(config, args):
......
...@@ -167,8 +167,9 @@ def check(audio_file): ...@@ -167,8 +167,9 @@ def check(audio_file):
def main_sp(config, args): def main_sp(config, args):
exp = DeepSpeech2Tester_hub(config, args) exp = DeepSpeech2Tester_hub(config, args)
exp.setup() with exp.eval():
exp.run_test() exp.setup()
exp.run_test()
def main(config, args): def main(config, args):
......
...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments ...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments
def main_sp(config, args): def main_sp(config, args):
exp = Tester(config, args) exp = Tester(config, args)
exp.setup() with exp.eval():
exp.run_align() exp.setup()
exp.run_align()
def main(config, args): def main(config, args):
......
...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments ...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments
def main_sp(config, args): def main_sp(config, args):
exp = Tester(config, args) exp = Tester(config, args)
exp.setup() with exp.eval():
exp.run_export() exp.setup()
exp.run_export()
def main(config, args): def main(config, args):
......
...@@ -24,8 +24,9 @@ from deepspeech.utils.utility import print_arguments ...@@ -24,8 +24,9 @@ from deepspeech.utils.utility import print_arguments
def main_sp(config, args): def main_sp(config, args):
exp = Tester(config, args) exp = Tester(config, args)
exp.setup() with exp.eval():
exp.run_test() exp.setup()
exp.run_test()
def main(config, args): def main(config, args):
......
...@@ -29,14 +29,14 @@ model_test_alias = { ...@@ -29,14 +29,14 @@ model_test_alias = {
def main_sp(config, args): def main_sp(config, args):
class_obj = dynamic_import(args.model_name, model_test_alias) class_obj = dynamic_import(args.model_name, model_test_alias)
exp = class_obj(config, args) exp = class_obj(config, args)
exp.setup() with exp.eval():
exp.setup()
if args.run_mode == 'test': if args.run_mode == 'test':
exp.run_test() exp.run_test()
elif args.run_mode == 'export': elif args.run_mode == 'export':
exp.run_export() exp.run_export()
elif args.run_mode == 'align': elif args.run_mode == 'align':
exp.run_align() exp.run_align()
def main(config, args): def main(config, args):
......
...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments ...@@ -20,8 +20,9 @@ from deepspeech.utils.utility import print_arguments
def main_sp(config, args): def main_sp(config, args):
exp = Tester(config, args) exp = Tester(config, args)
exp.setup() with exp.eval():
exp.run_export() exp.setup()
exp.run_export()
def main(config, args): def main(config, args):
......
...@@ -24,8 +24,9 @@ from deepspeech.utils.utility import print_arguments ...@@ -24,8 +24,9 @@ from deepspeech.utils.utility import print_arguments
def main_sp(config, args): def main_sp(config, args):
exp = Tester(config, args) exp = Tester(config, args)
exp.setup() with exp.eval():
exp.run_test() exp.setup()
exp.run_test()
def main(config, args): def main(config, args):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册