mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

fix ReID trainer bugs

Browse Source
This commit is contained in:
dongshuilong 2021-06-03 19:28:32 +08:00
parent c784209355
commit 10c0fc4b73
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/train.py
View File

@ -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()