bug fix

2019-05-06 11:04:25 +01:00 · 2019-05-06 11:04:25 +01:00 · 12f9fec960
parent 024913f73c
commit 12f9fec960
1 changed files with 1 additions and 1 deletions
--- a/scripts/main.py
+++ b/scripts/main.py
@ -64,7 +64,7 @@ def build_engine(args, datamanager, model, optimizer, scheduler):
                pooling_method=args.pooling_method
            )
        else:
-            engine = torchreid.engine.ImageTripletEngine(
+            engine = torchreid.engine.VideoTripletEngine(
                datamanager,
                model,
                optimizer,