Jerry Jiarui XU
|
61e1d5c814
|
[Enhancement] Support ignore_index for sigmoid BCE (#210)
* [Enhancement] Add args check for ignore_index
* Support ignore_index
|
2020-11-17 16:14:03 +08:00 |
|
Jerry Jiarui XU
|
5e264c608c
|
Generalized OHEM (#54)
* Generalized OHEM
* remove config
* update docstring
* fixed sort prob
* fixed valid_mask
|
2020-08-09 23:49:23 +08:00 |
|
Jerry Jiarui XU
|
1765c12985
|
Support FP16 (#21)
* Support FP16
* add miss folder
* add tests
* remove useless config
* update memory
* reduce config
* migrate fp16 to mmcv
* add model link
|
2020-07-20 15:17:18 +08:00 |
|
Jerry Jiarui XU
|
b975d3b72a
|
Add test tutorial (#9)
* add test tutorial
* remote torch/torchvision from requirements
* update getting started
* rename drop_out_ratio -> dropout_ratio
|
2020-07-12 23:53:56 +08:00 |
|
Jiarui XU
|
b2724da80b
|
init commit
|
2020-07-10 02:39:01 +08:00 |