Jerry Jiarui XU
|
9849a8dc23
|
[Refactor]: Unified parameter initialization (#567)
* [Refactor]: Unified parameter initialization
* fixed pretrained
|
2021-06-16 21:41:29 -07:00 |
|
Jerry Jiarui XU
|
ffcdfd36d0
|
[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
|
99e3e5e499
|
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
|
e1f5b9231e
|
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
|
2fc821ec13
|
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 |