• 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
test_hooks.py 2.74 KB