Commit Graph

8 Commits (d7f82e5dc873a1c9ab401128cabc2e1244188515)

Author SHA1 Message Date
MengzhangLI 0fd3972c41 [Feature] Support multiple losses during training (#818)
* multiple losses

* fix lint error

* fix typos

* fix typos

* Adding Attribute

* Fixing loss_ prefix

* Fixing loss_ prefix

* Fixing loss_ prefix

* Add Same

* loss_name must has 'loss_' prefix

* Fix unittest

* Fix unittest

* Fix unittest

* Update mmseg/models/decode_heads/decode_head.py

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-24 15:08:28 +08:00
Junjun2016 2fe0bddf5e [Dcos] Add header for files (#796)
* Add header for files

* Delete header in config files
2021-08-16 23:16:55 -07:00
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