diff --git a/tools/program.py b/tools/program.py index 18fead8ba81072202cf1ae13f32b21522a5e257f..2f8b0e4adb0a3473c386845c0c521d034a9f2c8f 100755 --- a/tools/program.py +++ b/tools/program.py @@ -166,7 +166,6 @@ def train(config, epoch_num = config['Global']['epoch_num'] print_batch_step = config['Global']['print_batch_step'] eval_batch_step = config['Global']['eval_batch_step'] - profiler_options = config['profiler_options'] global_step = 0 diff --git a/tools/train.py b/tools/train.py index ee81961414c2321373c9d18061cb2d1daf4d8b98..05d295aa99718c25b94a123c23d08c2904fe8c6a 100755 --- a/tools/train.py +++ b/tools/train.py @@ -127,8 +127,6 @@ def test_reader(config, device, logger): if __name__ == '__main__': - config, device, logger, vdl_writer = program.preprocess( - is_train=True) - logger.info(f"config.profiler_options: {config.profiler_options}") + config, device, logger, vdl_writer = program.preprocess(is_train=True) main(config, device, logger, vdl_writer) # test_reader(config, device, logger)