6300bd756e
Bugfix for cls_layer
...
In `any_softmax`, all operations are in-place, so pass into the `logits.clone()` to prevent outside logits changed.
2021-05-31 17:32:24 +08:00
0c8e3d9805
update imbalanced sampler
...
Summary: add a new sampler, which is useful for imbalanced or long-tail dataset. This refers to ufoym/imbalanced-dataset-sampler.
2021-04-21 17:05:10 +08:00
44cee30dfc
update fastreid v1.2
...
Summary:
1. refactor dataloader and heads
2. bugfix in fastattr, fastclas, fastface and partialreid
3. partial-fc supported in fastface
2021-04-02 21:33:13 +08:00
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
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
15e1729a27
update fastreid V1.0
2021-01-18 11:36:38 +08:00
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
4d573b8107
refactor reid head
...
Summary: merge BNneckHead, LinearHead and ReductionHead into EmbeddingHead
because they are highly similar and can be prepared for ClsHead
2020-09-10 10:57:37 +08:00
liaoxingyu