KaiyangZhou
|
54b92ee617
|
check if isinstance(model, nn.DataParallel)
|
2018-11-08 23:54:26 +00:00 |
|
KaiyangZhou
|
2bc29e37f5
|
fix bug
|
2018-11-08 23:09:33 +00:00 |
|
KaiyangZhou
|
205df55987
|
add fixbase functions (debug mode)
|
2018-11-08 23:04:50 +00:00 |
|
KaiyangZhou
|
cac6d72a6d
|
minor change: data_source (Dataset) -> data_source (list)
|
2018-11-08 22:34:24 +00:00 |
|
KaiyangZhou
|
9980b5378b
|
add support for htri training
|
2018-11-08 22:02:53 +00:00 |
|
KaiyangZhou
|
131b476439
|
update variable names
|
2018-11-08 22:02:22 +00:00 |
|
KaiyangZhou
|
9de0dd10b5
|
update args.py: add train_sampler; pool -> pool-tracklet-features
|
2018-11-08 22:01:47 +00:00 |
|
KaiyangZhou
|
8230cc5e65
|
test.py -> debug.py
|
2018-11-08 21:43:36 +00:00 |
|
KaiyangZhou
|
b16f175d7d
|
add test.py to gitignore
|
2018-11-08 21:42:32 +00:00 |
|
KaiyangZhou
|
57682b768e
|
add ranklogger; update variable names
|
2018-11-08 21:41:32 +00:00 |
|
KaiyangZhou
|
01fe76cf27
|
sample -> sample_method
|
2018-11-08 21:40:44 +00:00 |
|
KaiyangZhou
|
e17bbcddf3
|
source/target -> source_names/target_names
|
2018-11-08 21:40:18 +00:00 |
|
KaiyangZhou
|
8bfdbe24d1
|
args.sample -> args.sample_method; source -> source_names; target -> target_names
|
2018-11-08 21:39:19 +00:00 |
|
KaiyangZhou
|
c32cf80a9f
|
add RankLogger
|
2018-11-08 21:38:22 +00:00 |
|
KaiyangZhou
|
fdc451c0e5
|
add linear decay
|
2018-11-07 22:02:23 +00:00 |
|
KaiyangZhou
|
2d32f5b28a
|
add warnings
|
2018-11-07 21:55:04 +00:00 |
|
KaiyangZhou
|
e3ae601ffc
|
mv KeyError to ValueError
|
2018-11-07 21:54:46 +00:00 |
|
KaiyangZhou
|
893152a335
|
adapt to changes in data_manager
|
2018-11-07 21:46:39 +00:00 |
|
KaiyangZhou
|
7b47e645e9
|
add property
|
2018-11-07 21:46:23 +00:00 |
|
KaiyangZhou
|
1f1f01669b
|
change dataset to (query, gallery)
|
2018-11-07 21:45:48 +00:00 |
|
KaiyangZhou
|
137ef37df0
|
rm last comma
|
2018-11-07 20:49:51 +00:00 |
|
KaiyangZhou
|
3f9e20a55e
|
change access to dataloaders in dm by class method
|
2018-11-07 20:48:21 +00:00 |
|
KaiyangZhou
|
506369f9ad
|
correct test_batch to test_batch_size
|
2018-11-07 17:13:39 +00:00 |
|
KaiyangZhou
|
e6464f3443
|
update args for dataset/optimizer
|
2018-11-07 17:09:23 +00:00 |
|
KaiyangZhou
|
b81c13fd10
|
fix bug: query -> gallery
|
2018-11-07 15:54:10 +00:00 |
|
KaiyangZhou
|
1b255992bb
|
update dataset to dm
|
2018-11-07 15:40:02 +00:00 |
|
KaiyangZhou
|
2ecd8e5704
|
wrap argument_parser in args.py
|
2018-11-07 15:36:49 +00:00 |
|
KaiyangZhou
|
bc13a9fdc1
|
change BENCHMARK.md to MODEL_ZOO.md
|
2018-11-06 15:21:48 +00:00 |
|
KaiyangZhou
|
1cefe7b5f1
|
update data_manager to datasets
|
2018-11-06 13:52:57 +00:00 |
|
KaiyangZhou
|
7587dfe7ec
|
fix typo: criterion -> criterion_xent
|
2018-11-05 21:38:58 +00:00 |
|
KaiyangZhou
|
a96dd6b916
|
support multi-dataset training
|
2018-11-05 21:36:12 +00:00 |
|
KaiyangZhou
|
81d3cc3995
|
fixed division error: / -> //
|
2018-11-05 21:30:40 +00:00 |
|
KaiyangZhou
|
9726b8bb96
|
set required=True for dataset arguments
|
2018-11-05 21:25:09 +00:00 |
|
KaiyangZhou
|
02bac86c71
|
add data_manager.py
|
2018-11-05 21:19:27 +00:00 |
|
KaiyangZhou
|
ee44188b74
|
support multi-dataset training
|
2018-11-05 21:18:56 +00:00 |
|
KaiyangZhou
|
e96ac0e3a4
|
add warning to non-cython evaluation
|
2018-11-05 19:24:20 +00:00 |
|
KaiyangZhou
|
df0ae6d0e0
|
add num_cams; create base classes
|
2018-11-04 22:59:46 +00:00 |
|
KaiyangZhou
|
01d7594011
|
make camid start from 0
|
2018-11-04 21:27:08 +00:00 |
|
KaiyangZhou
|
c12b29bbfb
|
update requirements: torch>=0.4.1
|
2018-11-04 19:06:17 +00:00 |
|
KaiyangZhou
|
39df4855c1
|
add mlfn
|
2018-11-04 00:18:02 +00:00 |
|
KaiyangZhou
|
733d967345
|
rm multi-crop
|
2018-11-02 11:39:19 +00:00 |
|
KaiyangZhou
|
c388c24880
|
wrap transforms into build_transforms()
|
2018-11-01 22:29:19 +00:00 |
|
KaiyangZhou
|
e302af017d
|
recover dropout in pcb
|
2018-11-01 17:28:15 +00:00 |
|
KaiyangZhou
|
a57558738e
|
generalize CrossEntropyLableSmooth to CrossEntropyLoss
|
2018-10-29 23:00:17 +00:00 |
|
KaiyangZhou
|
752a4fe498
|
update pcb: commented dropout
|
2018-10-28 22:30:19 +00:00 |
|
KaiyangZhou
|
7588c8f581
|
update pcb: add support to various resnet archs
|
2018-10-28 11:44:09 +00:00 |
|
KaiyangZhou
|
12cdbf96b5
|
update pcb: mv conv5 after eval-mode in forward
|
2018-10-28 11:30:35 +00:00 |
|
KaiyangZhou
|
bfba6389d6
|
add pcb
|
2018-10-28 11:16:42 +00:00 |
|
KaiyangZhou
|
f77b9ada10
|
fill space after comma
|
2018-10-27 23:22:23 +01:00 |
|
KaiyangZhou
|
bc942aaf09
|
add variable model_url
|
2018-10-27 22:50:22 +01:00 |