fast-reid/fastreid/layers
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
..
__init__.py v0.3 update 2020-12-07 14:19:20 +08:00
activation.py onnx/trt support 2020-07-29 17:43:39 +08:00
arc_softmax.py refactor model arch 2020-09-01 16:14:45 +08:00
batch_drop.py refactor(fastreid) 2020-04-20 10:59:29 +08:00
batch_norm.py v0.3 update 2020-12-07 14:19:20 +08:00
circle_softmax.py refactor model arch 2020-09-01 16:14:45 +08:00
context_block.py refactor(fastreid) 2020-04-20 10:59:29 +08:00
cos_softmax.py v0.3 update 2020-12-07 14:19:20 +08:00
frn.py refactor(fastreid) 2020-04-20 10:59:29 +08:00
gather_layer.py fix triplet loss backward propagation on multi-gpu training (#82) 2020-09-28 17:16:51 +08:00
non_local.py v0.3 update 2020-12-07 14:19:20 +08:00
pooling.py extent gem pool 2020-09-10 11:01:22 +08:00
se_layer.py refactor(fastreid) 2020-04-20 10:59:29 +08:00
splat.py v0.3 update 2020-12-07 14:19:20 +08:00