From 61a1db566d73b4c1b029447c209eebd8e2215633 Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Wed, 14 Nov 2018 22:33:00 +0000 Subject: [PATCH] update --fixbase to --always-fixbase --- args.py | 4 ++-- train_imgreid_xent.py | 2 +- train_imgreid_xent_htri.py | 2 +- train_vidreid_xent.py | 2 +- train_vidreid_xent_htri.py | 2 +- 5 files changed, 6 insertions(+), 6 deletions(-) diff --git a/args.py b/args.py index f8e8893..17c5f98 100644 --- a/args.py +++ b/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'], diff --git a/train_imgreid_xent.py b/train_imgreid_xent.py index d25f086..fdb8619 100755 --- a/train_imgreid_xent.py +++ b/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) diff --git a/train_imgreid_xent_htri.py b/train_imgreid_xent_htri.py index 96ef9bd..e204450 100755 --- a/train_imgreid_xent_htri.py +++ b/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) diff --git a/train_vidreid_xent.py b/train_vidreid_xent.py index 6eca7d1..f5307a4 100755 --- a/train_vidreid_xent.py +++ b/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) diff --git a/train_vidreid_xent_htri.py b/train_vidreid_xent_htri.py index d8299f2..98b29b7 100755 --- a/train_vidreid_xent_htri.py +++ b/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)