Commit Graph

11 Commits (19f39539de7b26340ebcc394f0c89522b3536314)

Author SHA1 Message Date
Mladen Fernežir 9e17a0d364 Fix validation loss logging () 2022-04-22 15:25:32 +08:00
jbwang1997 44d4c3877e [Fix] Set random seed for generating `palette` if not given. ()
* Fix colors

* fix comments

* Add comments

* Add comments. Add random seed in datasets
2021-12-23 09:51:50 +08:00
Han Zhang e06cbcd8b1 [Fix] Fix dist training infinite waiting issue ()
* [] fix dist training infinite waiting issue

* print log_vars keys in assertion msg

* linting issue
2021-12-08 21:02:41 +08:00
Julius Zhang 96b369bd55 [Fix] Fix loss parse in val_step ()
* [Fix] Fix loss parse in val_step

* Add val_step unittest

* Add train_step unittest
2021-09-25 18:17:40 -07:00
Junjun2016 2fe0bddf5e [Dcos] Add header for files ()
* 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 ()
* [Refactor]: Unified parameter initialization

* fixed pretrained
2021-06-16 21:41:29 -07:00
Ziyi Wu ce56e68d30 [Enhance] Replace data_dict calling 'img' key to support MMDet3D ()
* remove dict calling img key for compatibility

* fix unit test

* infer batch size using len(result) to be consistent with mmcv
2021-04-29 01:01:34 -07:00
David de la Iglesia Castro 9cbb4b1288 Add opacity option to show_result () 2021-03-22 20:34:38 -07:00
ycr a932cbe02e Fix palette type. () 2020-07-23 13:01:31 +08:00
Jerry Jiarui XU e1f5b9231e Support FP16 ()
* 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
Jiarui XU b2724da80b init commit 2020-07-10 02:39:01 +08:00