1. 30 Jul, 2020 1 commit
    • shilong's avatar
      Ema (#421) · 1830347f
      shilong authored
      * add ema hook
      
      * add ema hook resume
      
      * add ema hook test
      
      * fix typo
      
      * fix according to comment
      
      * delete logger
      
      * fix according to comment
      
      * fix unitest
      
      * fix typo
      
      * fix according to comment
      
      * change to resume_from
      
      * typo
      
      * fix isort
      1830347f
  2. 25 Jul, 2020 1 commit
    • Wang Xinjiang's avatar
      Syncbuf (#447) · d4da3daa
      Wang Xinjiang authored
      * More robust sync buffer hook
      
      * More robust sync buffer hook
      
      * Reformat
      d4da3daa
  3. 24 Jul, 2020 1 commit
  4. 17 Jul, 2020 1 commit
  5. 03 Jul, 2020 1 commit
  6. 15 Jun, 2020 1 commit
    • Harry's avatar
      Add CosineRestartLrUpdaterHook (#319) · f28a7c7e
      Harry authored
      * 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
      f28a7c7e
  7. 02 Jun, 2020 1 commit
  8. 26 Apr, 2020 1 commit
  9. 22 Apr, 2020 1 commit
  10. 19 Apr, 2020 1 commit
    • Wenwei Zhang's avatar
      Momentum scheduler (#167) · ba059611
      Wenwei Zhang authored
      * track progress of iter&enum
      
      * restore
      
      * add momentum scheduler
      
      * fix small bug
      
      * cyclic scheduler"
      
      * fix bug
      
      * fix second phase's bug
      
      * reformat
      
      * feature (cosine lr): use relative ratio for more flexible scheduler
      
      * Fix (runner): fix bugs in runner
      
      * Refactor (hook): refactor cosing/cyclic LR/momentum hook with unittest
      
      * Clean unnecessary files and reformat
      
      * Fix memory key error when GPU is not avaliable
      
      * Resolve comments
      
      * Do not print momentum in text log
      
      * Change hook register order
      
      * Refactor max_iter
      
      * Fix max_iter bugs in runner
      
      * Enforce target_ratio to be either tuple or float
      ba059611
  11. 14 Apr, 2020 1 commit
    • David de la Iglesia Castro's avatar
      Add MlflowLoggerHook (#221) · d5f190d1
      David de la Iglesia Castro authored
      * Add MLflowLoggerHook
      
      * Add MLflowLoggerHook to __all__
      
      * Update name
      
      * Fix tracking.MlflowClient setup
      
      * Fix log_metric
      
      * Fix mlflow_pytorch import
      
      * Handle active_run
      
      * Fix self.mlflow reference
      
      * Simplify using high level API
      
      * Fix set_experiment
      
      * Add only_if_torch_available decorator and test_mlflow_hook
      
      * Add missing import in hooks
      
      * Fix torch available check
      
      * Patch mlflow.pytorch in test
      
      * Parametrize log_model
      
      * Fix log_model parametrize
      
      * Add docstring
      
      * Move wand patch
      
      * Fix flake8
      
      * Add regression test for non numeric metric
      
      * Only log numbers
      
      * Rename experiment_name-> exp_name
      
      * Remove pytest skip
      d5f190d1
  12. 16 Mar, 2020 1 commit
  13. 02 Mar, 2020 1 commit