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
b507fc207b
[Fix] Use MMCV's EvalHook in MMClassification ( #182 )
...
* use eval_hook in mmcv
* minor fix
2021-03-25 17:38:51 +08:00
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
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
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
fbc68bf266
Add classifiers, heads, necks and losses
2020-07-07 19:32:06 +08:00
70cda9c75e
fix linting
2020-05-21 21:27:54 +08:00
a99e6eb10d
init commit
2020-05-21 21:21:43 +08:00
WRH
LXXXXR
Lei Yang
David de la Iglesia Castro
yanglei
chenkai