Kai Chen
af02ac9f01
Use tmp dir as work_dir of runner ( #236 )
...
* use tmp dir as work_dir of runner
* only run codecov for python 3.7
* remove useless comments
2020-04-22 23:33:54 +08:00
David de la Iglesia Castro
d5f190d12d
Add MlflowLoggerHook ( #221 )
...
* 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
2020-04-14 23:54:55 +08:00
Kai Chen
12e5913bb9
Remove supports for python 2.7 and bump version to 0.4.0 ( #211 )
...
* remove supports for python 2.7
* fix the unit test for python 3.5
* add python 3.8 in CI
* try ubuntu 18.04 as the environment
2020-03-16 15:15:35 +08:00
Kai Chen
3fa5851891
Update pre-commit hook config ( #200 )
...
* update pre-commit hook config
* update the contributing guide
2020-03-01 15:51:40 +08:00
Kai Chen
c2c9fced2b
Add logging utils ( #196 )
...
* add logging utils
* install torch (cpu only) for CI
* fix unittests for runner
* remove python 2.7 from testing list
* add a corner case
2020-02-24 22:31:36 +08:00
Kai Chen
10fa1eeaf4
Add copyright header ( #171 )
...
* add copyright header
* change # to // for cpp files
2020-01-10 13:34:42 +08:00
Wenwei Zhang
da7bb063d3
Load state dict ( #164 )
...
* use load_from_state_dict to load ckpt
* reformat
* reformat
* reformat import
* pass flake8
* isort skip
2020-01-04 14:55:01 +08:00
Vladimir Iglovikov
56971278db
Added Weight and Biases logger. ( #143 )
...
* Added wandb logger
* bugfix
* fixes
* fix
* flake8 fixes
* yapf fixes
* Refactor hook, add test
* flake8 fixes
* yapf fixes
* fix in test
* mock package => travis
2019-12-02 12:32:50 +08:00
Kai Chen
eae81c1e86
Add yapf and isort to travis ( #96 )
...
* add yapf and isort to travis
* minor formatting
* remove blank lines
* skip unit tests for progressbar when python2
* update travis to ubuntu 16.04
* use a newer version ffmpeg
* add -y to add-apt-repository
2019-07-30 23:15:56 +08:00
Evgeny Nizhibitsky
395efbf2e9
Make latest checkpoint symlink relative ( #75 )
...
* Make latest symlink relative. Add tests
* Skip test_save_checkpoint in the absense of torch
* Fix pep8 in test_runner
* Import runner only if torch import has succeeded
2019-06-15 19:56:27 +08:00
