提交 c119c9c5 编写于 作者: L LielinJiang 提交者: lvmengsi

fix ema model clean error (#3404)

上级 742ae3b4
......@@ -41,7 +41,9 @@ def main():
item_clean = item.replace('_ema_0', '')
shutil.copyfile(os.path.join(args.ema_model_dir, item),
os.path.join(args.cleaned_model_dir, item_clean))
elif item.find('mean') > -1 or item.find('variance') > -1:
shutil.copyfile(os.path.join(args.ema_model_dir, item),
os.path.join(args.cleaned_model_dir, item))
if __name__ == '__main__':
main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册