1. 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
  2. 02 Jun, 2020 1 commit
  3. 26 Apr, 2020 1 commit
  4. 22 Apr, 2020 1 commit
  5. 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
  6. 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
  7. 16 Mar, 2020 1 commit
  8. 02 Mar, 2020 1 commit