Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to integrate mlflow with tensorflow object detection api

I am trying to use the mlflow.tensorflow.autolog() with tensorflow object detection api. Adding mlflow.tensorflow.autolog() to model_main.py logs some parameters like global_norm/clipped_gradient_norm,global_norm/gradient_norm,global_step/sec,learning_rate_1,loss_1,loss_2 in mlflow. However the more important metrics like map,precision,recall are not being logged in mlfow.

like image 202
zishan ahmed Avatar asked Dec 11 '25 02:12

zishan ahmed


1 Answers

The mlflow autologging will log whatever metrics are specified for the keras model, e.g. as in https://keras.io/metrics/ and https://github.com/mlflow/mlflow/blob/master/examples/keras/train.py#L66. Are those metrics specified but not logged?

like image 74
Sue Avatar answered Dec 13 '25 16:12

Sue



Donate For Us

If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!