Update main.py

ShengcaiLiao · Mar 31, 2021 · 06d7473 · 06d7473
1 parent e057350
commit 06d7473
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -329,7 +329,7 @@ def main(args):
                         help="number of channels for the final neck layer, default: 64")
     parser.add_argument('--ibn', type=str, choices={'a', 'b'}, default=None, help="IBN type. Choose from 'a' or 'b'. Default: None")
     # TLift
-    parser.add_argument('--do_tlift', action='store_false', default=True, help="apply TLift, default: True")
+    parser.add_argument('--do_tlift', action='store_true', default=False, help="apply TLift, default: False")
     parser.add_argument('--tau', type=float, default=100,
                         help="the interval threshold to define nearby persons in TLift, default: 100")
     parser.add_argument('--sigma', type=float, default=200,