diff --git a/scripts/main.py b/scripts/main.py index 33433df..a5c90a6 100755 --- a/scripts/main.py +++ b/scripts/main.py @@ -151,13 +151,16 @@ def main(): if args.resume and check_isfile(args.resume): args.start_epoch = resume_from_checkpoint(args.resume, model, scheduler=scheduler, optimizer=optimizer) - + # lr changing is BUG-gy! if args.lr != optimizer.param_groups[0]['lr']: old_lr = optimizer.param_groups[0]['lr'] change_lr_to(optimizer,args.lr) print('Changed optimzer lr from {} to {}.'.format(old_lr,args.lr)) + # NOTE: not resuming scheduler + scheduler = torchreid.optim.build_lr_scheduler(optimizer, **lr_scheduler_kwargs(args)) + print('Building {}-engine for {}-reid'.format(args.loss, args.app)) engine = build_engine(args, datamanager, model, optimizer, scheduler, experiment) diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-2/events.out.tfevents.1562676350.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-2/events.out.tfevents.1562676350.michael-manjaro-desktop new file mode 100644 index 0000000..e69de29 diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-1 /events.out.tfevents.1562677138.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-1 /events.out.tfevents.1562677138.michael-manjaro-desktop new file mode 100644 index 0000000..d84a5d9 Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-1 /events.out.tfevents.1562677138.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-10 /events.out.tfevents.1562677138.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-10 /events.out.tfevents.1562677138.michael-manjaro-desktop new file mode 100644 index 0000000..6ff913a Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-10 /events.out.tfevents.1562677138.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-2 /events.out.tfevents.1562677138.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-2 /events.out.tfevents.1562677138.michael-manjaro-desktop new file mode 100644 index 0000000..0ee85ab Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-2 /events.out.tfevents.1562677138.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-20 /events.out.tfevents.1562677138.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-20 /events.out.tfevents.1562677138.michael-manjaro-desktop new file mode 100644 index 0000000..582015b Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-20 /events.out.tfevents.1562677138.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-3 /events.out.tfevents.1562677138.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-3 /events.out.tfevents.1562677138.michael-manjaro-desktop new file mode 100644 index 0000000..6fe2ea4 Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-3 /events.out.tfevents.1562677138.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-5 /events.out.tfevents.1562677138.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-5 /events.out.tfevents.1562677138.michael-manjaro-desktop new file mode 100644 index 0000000..4c0108f Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/eval/ranks/eval-rank-5 /events.out.tfevents.1562677138.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/events.out.tfevents.1562676389.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/events.out.tfevents.1562676389.michael-manjaro-desktop new file mode 100644 index 0000000..dd66035 Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/events.out.tfevents.1562676389.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-1/events.out.tfevents.1562676390.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-1/events.out.tfevents.1562676390.michael-manjaro-desktop new file mode 100644 index 0000000..992a8ff Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-1/events.out.tfevents.1562676390.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-2/events.out.tfevents.1562676390.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-2/events.out.tfevents.1562676390.michael-manjaro-desktop new file mode 100644 index 0000000..122aff1 Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-2/events.out.tfevents.1562676390.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-3/events.out.tfevents.1562676390.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-3/events.out.tfevents.1562676390.michael-manjaro-desktop new file mode 100644 index 0000000..7c4420d Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-3/events.out.tfevents.1562676390.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-4/events.out.tfevents.1562676390.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-4/events.out.tfevents.1562676390.michael-manjaro-desktop new file mode 100644 index 0000000..cc852de Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-4/events.out.tfevents.1562676390.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-5/events.out.tfevents.1562676390.michael-manjaro-desktop b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-5/events.out.tfevents.1562676390.michael-manjaro-desktop new file mode 100644 index 0000000..63c2f2b Binary files /dev/null and b/scripts/runs/osnet_custom-softmax-2048featdim-3/optim/train-accs/top-5/events.out.tfevents.1562676390.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_x1_0-softmax-2/events.out.tfevents.1562676053.michael-manjaro-desktop b/scripts/runs/osnet_x1_0-softmax-2/events.out.tfevents.1562676053.michael-manjaro-desktop new file mode 100644 index 0000000..94844b0 Binary files /dev/null and b/scripts/runs/osnet_x1_0-softmax-2/events.out.tfevents.1562676053.michael-manjaro-desktop differ diff --git a/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-1/events.out.tfevents.1562676055.michael-manjaro-desktop b/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-1/events.out.tfevents.1562676055.michael-manjaro-desktop new file mode 100644 index 0000000..e69de29 diff --git a/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-2/events.out.tfevents.1562676055.michael-manjaro-desktop b/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-2/events.out.tfevents.1562676055.michael-manjaro-desktop new file mode 100644 index 0000000..e69de29 diff --git a/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-3/events.out.tfevents.1562676055.michael-manjaro-desktop b/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-3/events.out.tfevents.1562676055.michael-manjaro-desktop new file mode 100644 index 0000000..e69de29 diff --git a/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-4/events.out.tfevents.1562676055.michael-manjaro-desktop b/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-4/events.out.tfevents.1562676055.michael-manjaro-desktop new file mode 100644 index 0000000..e69de29 diff --git a/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-5/events.out.tfevents.1562676055.michael-manjaro-desktop b/scripts/runs/osnet_x1_0-softmax-2/optim/train-accs/top-5/events.out.tfevents.1562676055.michael-manjaro-desktop new file mode 100644 index 0000000..e69de29 diff --git a/torchreid/models/osnet.py b/torchreid/models/osnet.py index b18de80..bafe09c 100644 --- a/torchreid/models/osnet.py +++ b/torchreid/models/osnet.py @@ -316,7 +316,7 @@ def forward(self, x): def osnet_custom(num_classes=1000, loss='softmax', **kwargs): # custom osnet return OSNet(num_classes, blocks=[OSBlock, OSBlock, OSBlock], layers=[2, 2, 2], - channels=[64, 256, 384, 512], feature_dim=1024, loss=loss, **kwargs) + channels=[64, 256, 384, 512], feature_dim=2048, loss=loss, **kwargs) def osnet_x1_0(num_classes=1000, loss='softmax', **kwargs): # standard size (width x1.0)