9 Commits (a47451b47a1c88668c8284c885be92b5965e5447)

Author	SHA1	Message	Date
Jintao Lin	1ebd7ea6fb	add unittest for set_random_seed (#376)	2020-07-02 00:13:04 +08:00
Harry	69048ff056	Specifying distributed training port in os.environ when training with slurm (#362) ... * feat: support for os.environ port for slurm training * fix: port data type * feat: add flawed unittest * feat: add flawed unittest * docs: add comments * fix: unittest * fix: unittest	2020-06-20 00:49:44 +08:00
Kai Chen	6bb244f255	add train_step() and val_step() for MMDP (#354)	2020-06-18 20:55:53 +08:00
Harry	f28a7c7ed7	Add CosineRestartLrUpdaterHook (#319) ... * feat: add CosineRestartLrUpdaterHook * style: rename period to periods * fix: bug in period 0 * feat: rename eta_min to min_lr and add min_lr_ratio * docs: fix docstring of restart lr updater * refactor: use annealing_cos * docs: add docstring to annealing_cos * feat: cosine restart lr update hook * refactor: modify code order for unittest	2020-06-15 23:01:26 +08:00
Harry	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
Kai Chen	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
Kai Chen	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
Wenwei Zhang	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
Kai Chen	a338d43d78	Refactor unittests (#241) ... * refactor unittests * split test_video.py to two files	2020-04-26 22:54:27 +08:00