fast-reid/fastreid/modeling/heads
liaoxingyu 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
..
__init__.py support classification in fastreid (#443) 2021-03-26 20:17:39 +08:00
build.py add configurable decorator & linear loss decouple (#441) 2021-03-23 12:10:06 +08:00
clas_head.py Bugfix for cls_layer 2021-05-31 17:32:24 +08:00
embedding_head.py Bugfix for cls_layer 2021-05-31 17:32:24 +08:00