Commit Graph

8 Commits (5ad3bed2cdb791184e45d883816e275a9855c4dc)

Author SHA1 Message Date
Ma Zerun c03efeeea4
[Feature] Support MPS device. (#894)
* [Feature] Support MPS device.

* Add `auto_select_device`

* Add unit tests
2022-07-28 12:28:51 +08:00
Ma Zerun 702c196514
[Docs] Refine API reference. (#774)
* [Docs] Refine API reference

* Add PoolFormer
2022-04-15 20:19:20 +08:00
Ma Zerun f552419e45
[Enhance] Add setup multi-processing both in train and test. (#671) 2022-01-27 10:18:36 +08:00
Ma Zerun 894a82ea4f
[Refactor] Change `load_json_logs` to `load_json_log`. (#602) 2021-12-14 18:03:27 +08:00
Ezra-Yu b486bb23f8
[Feature] Add a tool to visualize learning rate in each iterations (#498)
* add vis lr

* add doc

* Update doc and rm load_json_log in analysis

* Update docs

* add unit tests and update docstring

* fix unit-tests

* Use DummyIterBasedRunner and DummyEpochBasedRunner to simulate training
process.

* rm function val and add judge

* update docs

* improve english expression

* Update docs/tools/visualization.md

Co-authored-by: Ma Zerun <mzr1996@163.com>

* fix typo and ImageNet datasize

* fix typo

* update example

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-11-03 10:48:56 +08:00
Ma Zerun f9eb9b409b
[Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
LXXXXR 5de2d2c0e9
Change to use mmcv.utils.collect_env to avoid duplicate codes (#103) 2020-12-01 14:24:08 +08:00
chenkai a99e6eb10d init commit 2020-05-21 21:21:43 +08:00