提交 97a66874 编写于 作者: M MissPenguin

refine

上级 053cc43d
...@@ -338,8 +338,12 @@ def train(config, ...@@ -338,8 +338,12 @@ def train(config,
return return
def eval(model, valid_dataloader, post_process_class, eval_class, def eval(model,
model_type, use_srn=False): valid_dataloader,
post_process_class,
eval_class,
model_type,
use_srn=False):
model.eval() model.eval()
with paddle.no_grad(): with paddle.no_grad():
total_frame = 0.0 total_frame = 0.0
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册