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
Xingyu Liao
1dce15efad
faster dataloader with pre-fetch and cuda stream ( #456 )
...
Summary: add a background thread to create a generator with pre-fetch, and create a new cuda stream to copy tensor from cpu to gpu in parallel.
Reviewed by: l1aoxingyu
2021-04-12 15:03:35 +08:00
liaoxingyu
55300730e1
update fastreid v1.2 readme and changelog
2021-04-06 20:09:13 +08:00
liaoxingyu
1666c82db4
update readme about FastRT
2021-01-30 16:11:08 +08:00
liaoxingyu
77a91b1204
feat: support multi-teacher kd
...
Summary: support multi-teacher kd with logits and overhaul distillation
2021-01-29 17:25:31 +08:00