trainer eval when not train

pull/910/head
Hui Zhang 4 years ago
parent fa5531c03e
commit 97136fe97f

@ -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
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
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):
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
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
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
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 = {
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
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
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):

Loading…
Cancel
Save