提交 10c0fc4b 编写于 作者: D dongshuilong

fix ReID trainer bugs

上级 c7842093
......@@ -22,9 +22,13 @@ sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
from ppcls.utils import config
from ppcls.engine.trainer import Trainer
from ppcls.engine.trainer_reid import TrainerReID
if __name__ == "__main__":
args = config.parse_args()
config = config.get_config(args.config, overrides=args.override, show=True)
trainer = Trainer(config, mode="train")
if "Trainer" in config:
trainer = eval(config["Trainer"]["name"])(config, mode="train")
else:
trainer = Trainer(config, mode="train")
trainer.train()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册