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

fix trainer

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