7730a79fcd
fix typo of annealing ( #433 )
2020-07-17 23:48:22 +08:00
6bb244f255
add train_step() and val_step() for MMDP ( #354 )
2020-06-18 20:55:53 +08:00
67a26da917
Add IterBasedRunner ( #314 )
...
* feat: add IterBasedRunner
* fix: unittest
* feat: more unittest
* fix: expose dataloader len
* minor updates of BaseRunner
* refactor: remove CosineRestartLrUpdaterHook
* style: add docstring
* refactor: update IterTextLoggerHook: fstring and exp_name
* fix: epoch_runner unittest
* refactor: remove IterBasedTextLogger
* fix: old IterTextLoggerHook issue
* refactor: remove __len__ of IterLoader
* feat: add IterBasedRunner to init
* feat: add __len__ to IterLoader
* fix some docstrings
* refactor: use is_parallel_module
* fix: import issue
* fix: runner unittest missing logger
* fix checkpoints
* feat: add by_epoch default value to IterBaseRunner regitering loggger_hook
* refactor: remove setting by_epoch in log_config
* minor refactoring
* docs: add docstring
* fix: remove unused doc
* update the log info for saving checkpoints
Co-authored-by: Kai Chen <chenkaidev@gmail.com>
2020-06-11 13:35:34 +08:00
821b3ad622
Fix the BC issue of ddp ( #325 )
...
* fix the BC issue of ddp
* minor fix for the docstring
2020-06-08 22:34:19 +08:00
35ba152821
Add a BaseRunner and rename Runner to EpochBasedRunner ( #290 )
...
* add a BaseRunner and rename Runner to EpochBasedRunner
* fix the train/val step
* bug fix
* update unit tests
* fix unit tests
* raise an error if both batch_processor and train_step are set
* add a unit test
2020-06-02 22:23:21 +08:00
19e4a06cbc
Fix CosineAnealingLr register bug ( #265 )
...
* Fall back to CosineLr
* Fix consineanealing with unittest
* Cover momentum hook
* Add comments to explain
2020-05-04 00:38:55 +08:00
a338d43d78
Refactor unittests ( #241 )
...
* refactor unittests
* split test_video.py to two files
2020-04-26 22:54:27 +08:00
Yawei Li
Kai Chen
Harry
Wenwei Zhang