mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
fast-reid/fastreid/layers
History
liaoxingyu 16655448c2 onnx/trt support 
Summary: change model pretrain mode and support onnx/TensorRT export
 		2020-07-29 17:43:39 +08:00
..
__init__.py onnx/trt support 2020-07-29 17:43:39 +08:00
activation.py onnx/trt support 2020-07-29 17:43:39 +08:00
am_softmax.py onnx/trt support 2020-07-29 17:43:39 +08:00
arc_softmax.py onnx/trt support 2020-07-29 17:43:39 +08:00
batch_drop.py refactor(fastreid) 2020-04-20 10:59:29 +08:00
batch_norm.py finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
circle_softmax.py onnx/trt support 2020-07-29 17:43:39 +08:00
context_block.py refactor(fastreid) 2020-04-20 10:59:29 +08:00
frn.py refactor(fastreid) 2020-04-20 10:59:29 +08:00
non_local.py feat: add SyncBN and GroupNorm suppor 2020-05-14 11:36:28 +08:00
pooling.py update fast global avgpool 2020-06-12 16:34:03 +08:00
se_layer.py refactor(fastreid) 2020-04-20 10:59:29 +08:00
splat.py onnx/trt support 2020-07-29 17:43:39 +08:00