Skip to content

GitLab

Commit 40023f76 authored by WenmuZhou's avatar WenmuZhou
Browse files

修复ips和batch_cost输出错误的bug

parent 0c287c41
Show whitespace changes
Inline Side-by-side
Showing with 14 additions and 12 deletions
tools/program.py
View file @ 40023f76
...@@ -236,7 +236,6 @@ def train(config, ...@@ -236,7 +236,6 @@ def train(config,
train_batch_cost = 0.0 train_batch_cost = 0.0
train_reader_cost = 0.0 train_reader_cost = 0.0
batch_sum = 0 batch_sum = 0
batch_start = time.time()
# eval # eval
if global_step > start_eval_step and \ if global_step > start_eval_step and \
(global_step - start_eval_step) % eval_batch_step == 0 and dist.get_rank() == 0: (global_step - start_eval_step) % eval_batch_step == 0 and dist.get_rank() == 0:
...@@ -275,6 +274,7 @@ def train(config, ...@@ -275,6 +274,7 @@ def train(config,
best_model_dict[main_indicator], best_model_dict[main_indicator],
global_step) global_step)
global_step += 1 global_step += 1
batch_start = time.time()
if dist.get_rank() == 0: if dist.get_rank() == 0:
save_model( save_model(
model, model,
...@@ -334,7 +334,9 @@ def eval(model, valid_dataloader, post_process_class, eval_class): ...@@ -334,7 +334,9 @@ def eval(model, valid_dataloader, post_process_class, eval_class):
def save_inference_mode(model, config, logger): def save_inference_mode(model, config, logger):
if dist.get_rank() == 0:
model.eval() model.eval()
print('infer')
save_path = '{}/infer/{}'.format(config['Global']['save_model_dir'], save_path = '{}/infer/{}'.format(config['Global']['save_model_dir'],
config['Architecture']['model_type']) config['Architecture']['model_type'])
if config['Architecture']['model_type'] == 'rec': if config['Architecture']['model_type'] == 'rec':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment