update --fixbase to --always-fixbase

args.py

@@ -86,8 +86,8 @@ def argument_parser():
     parser.add_argument('--test-batch-size', default=100, type=int,
                         help="test batch size")
     
-    parser.add_argument('--fixbase', action='store_true',
-                        help="always fix base network")
+    parser.add_argument('--always-fixbase', action='store_true',
+                        help="always fix base network and only train specified layers")
     parser.add_argument('--fixbase-epoch', type=int, default=0,
                         help="how many epochs to fix base network (only train randomly initialized classifier)")
     parser.add_argument('--open-layers', type=str, nargs='+', default=['classifier'],

train_imgreid_xent.py

@@ -157,7 +157,7 @@ def train(epoch, model, criterion, optimizer, trainloader, use_gpu, fixbase=Fals
 
     model.train()
 
-    if fixbase or args.fixbase:
+    if fixbase or args.always_fixbase:
         open_specified_layers(model, args.open_layers)
     else:
         open_all_layers(model)

train_imgreid_xent_htri.py

@@ -160,7 +160,7 @@ def train(epoch, model, criterion_xent, criterion_htri, optimizer, trainloader,
 
     model.train()
 
-    if fixbase or args.fixbase:
+    if fixbase or args.always_fixbase:
         open_specified_layers(model, args.open_layers)
     else:
         open_all_layers(model)

train_vidreid_xent.py

@@ -158,7 +158,7 @@ def train(epoch, model, criterion, optimizer, trainloader, use_gpu, fixbase=Fals
 
     model.train()
 
-    if fixbase or args.fixbase:
+    if fixbase or args.always_fixbase:
         open_specified_layers(model, args.open_layers)
     else:
         open_all_layers(model)

train_vidreid_xent_htri.py

@@ -162,7 +162,7 @@ def train(epoch, model, criterion_xent, criterion_htri, optimizer, trainloader,
 
     model.train()
 
-    if fixbase or args.fixbase:
+    if fixbase or args.always_fixbase:
         open_specified_layers(model, args.open_layers)
     else:
         open_all_layers(model)