liaoxingyu
07b8251ccb
Support gradient clip
...
Follow detectron2's instruction and add gradient clip in step function of optimizer
2021-05-31 17:11:37 +08:00
Xingyu Liao
fb36b23678
bugfix for attribute project ( #450 )
...
Summary: refactor sample weight in attribute recognition;
change all options to False in defaults.py and modify yaml files
2021-03-31 17:07:19 +08:00
Xingyu Liao
be0a089e1f
bugfix & merge classification transforms ( #448 )
...
Summary: change heads definition in project and config file, merge classification transforms into default transforms
2021-03-30 15:47:14 +08:00
Xingyu Liao
890224f25c
support classification in fastreid ( #443 )
...
Summary: support classification and refactor build_dataloader which can support explicit parameters passing
2021-03-26 20:17:39 +08:00
Xingyu Liao
883fd4aede
add configurable decorator & linear loss decouple ( #441 )
...
Summary: Add configurable decorator which can call `Baseline` with `Baseline(cfg)` or `Baseline(cfg, heads=heads, ...)`
Decouple linear and loss computation for partial-fc support.
Reviewed By: l1aoxingyu
2021-03-23 12:10:06 +08:00
liaoxingyu
e26182e6ec
make lr warmup by iter
...
Summary: change warmup way by iter not by epoch, which will make it more flexible when training small epochs
2021-01-22 11:17:21 +08:00
liaoxingyu
15e1729a27
update fastreid V1.0
2021-01-18 11:36:38 +08:00
liaoxingyu
a327a70f0d
v0.3 update
...
Summary:
1. change DPP training in apex way;
2. make warmup scheduler by iter and lr scheduler by epoch;
3. replace random erasing with torchvision implementation;
4. naming modification in config file
2020-12-07 14:19:20 +08:00
liaoxingyu
3c48bf78c1
change config file
2020-09-10 10:58:56 +08:00
liaoxingyu
16655448c2
onnx/trt support
...
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
liaoxingyu
825da4dca0
update all config files
2020-07-06 16:48:14 +08:00
liaoxingyu
eebc71b7c8
update config
...
Summary: remove use_cosine_dist in config which is equivalent to euclidean distance with norm feature
2020-06-16 11:14:58 +08:00
liaoxingyu
e20eb0394b
fix: modify open_layers in BoT config file
...
modify open_layers to list type to be compatible with default config
2020-05-19 14:46:35 +08:00
liaoxingyu
18a33f7962
feat: add MGN model
...
support MGN architecture and training config
2020-05-15 11:39:54 +08:00
liaoxingyu
46228ce946
chore(configs): update all training config
2020-05-01 09:04:51 +08:00
liaoxingyu
2327a5565f
chore(configs): update Market1501 training config
2020-04-29 21:06:57 +08:00
liaoxingyu
ec19bcc1d3
style(configs): put all config files together
...
put all config files into one place for easily control,
and add tools for put train_net.py which almost the same in
different projects
2020-04-29 16:18:54 +08:00