mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
501 Commits
8 Branches
41 Tags
32 MiB
Commit Graph

18 Commits (a7f8e96b31c10ab3e9c133293ca406e6e548475b)

Author SHA1 Message Date
Ma Zerun 8ef5aeda40
[Enhance] New-style CPU training and inference. (#674) 
* [Enhance] New-style CPU training and inference.

* Add version check in CPU training/test
 2022-01-30 20:49:54 +08:00
Zhicheng Chen da39ca6898
[Fix] Fix test sampler bug. (#611) 
* sampler bugfixes

* sampler bugfixes

* reorganize code

* minor fixes

* reorganize code

* minor fixes

* Use `mmcv.runner.get_dist_info` instead of `dist` package to get rank
and world size.

* Add `build_dataloader` unit tests and fix sampler's unit tests.

* Fix unit tests

* Fix unit tests

Co-authored-by: mzr1996 <mzr1996@163.com>
 2021-12-23 18:07:41 +08:00
Zhicheng Chen d5ae255061
[Feature] Repeated Aug and Sampler Registry (#588) 2021-12-21 15:26:28 +08:00
Ma Zerun d25a78d547
[Enchance] Set a random seed when the user does not set a seed. (#554) 2021-12-02 18:09:55 +08:00
Youqing Xiaozhua 1177feb05b
[Fix] Remove `DistSamplerSeedHook` if use `IterBasedRunner`. (#501) 
* remove DistSamplerSeedHook for IterBasedRunner

* Register DistSamplerSeedHook for EpochBasedRunner only

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

Co-authored-by: Ma Zerun <mzr1996@163.com>
 2021-10-27 17:07:33 +08:00
takuoko f68f17e9bb
[Fix] Set the priority of `EvalHook` to "LOW" to avoid a bug of `IterBasedRunner` (#488) 
* Set the priority of EvalHook to LOW.

* add a comment
 2021-10-20 16:24:52 +08:00
Tan Y.Q c1b3757ae4
[Fix] Fix a wrong paramter of `get_root_logger` in `apis/train.py`. (#486) 
The get_root_logger(log_file=None, log_level=logging.INFO) function does not needs the string type cfg.log_level as input. This bug does not have negative effect.
 2021-10-13 14:54:47 +08:00
Ma Zerun f9eb9b409b
[Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
whcao d62f198d2b
[Feature]Support custom hooks (#305) 
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* register custom_hooks in runner

* set custom_hooks_config to cfg.get(custom_hooks, None)
 2021-06-15 21:09:58 +08:00
WRH b99bd4fa88
Fix bug for CPU training (#286) 
* remove MMDataParallel when using cpu

* support cpu testing

* fix lint
 2021-06-12 22:26:33 +08:00
WRH 9be435846c
Support training on CPU (#219) 
* draft

* add parameter for training tools

* Update .pre-commit-config.yaml
 2021-04-26 13:56:45 +08:00
LXXXXR b507fc207b
[Fix] Use MMCV's EvalHook in MMClassification (#182) 
* use eval_hook in mmcv

* minor fix
 2021-03-25 17:38:51 +08:00
LXXXXR e76c5a368d
[Feature] Support fp16 training (#178) 
* change mmcls fp16 to mmcv hook

* support fp16

* clean unnessary stuff
 2021-03-17 15:53:55 +08:00
Lei Yang 784987fe9f
Fix slurm evaluation error during training (#69) 
* visualize results on image demo

* set round_up to True during test
 2020-10-20 15:04:39 +08:00
David de la Iglesia Castro 4f4f2957ef
Use build_runner (#54) 
* Use build_runner in train api

* Support iter in eval_hook

* Add runner section

* Add test_eval_hook

* Pin mmcv version in install docs

* Replace max_iters with max_epochs

* Set by_epoch=True as default

* Remove trailing space

* Replace DeprecationWarning with UserWarning

* pre-commit

* Fix tests
 2020-10-15 21:12:50 +08:00
yanglei fbc68bf266 Add classifiers, heads, necks and losses 2020-07-07 19:32:06 +08:00
chenkai 70cda9c75e fix linting 2020-05-21 21:27:54 +08:00
chenkai a99e6eb10d init commit 2020-05-21 21:21:43 +08:00