提交 89229c62 编写于 作者: littletomatodonkey's avatar littletomatodonkey

fix trainer

上级 868ae8af
......@@ -67,8 +67,6 @@ class Trainer(object):
dist.init_parallel_env()
if "Head" in self.config["Arch"]:
self.config["Arch"]["Head"]["class_num"] = self.config["Global"][
"class_num"]
self.is_rec = True
else:
self.is_rec = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册