mmpretrain

Commit Graph

Author	SHA1	Message	Date
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

13 Commits (52e62560d445c4428d6a54b111d4a4f486d0bc8c)