I am also having same error.

Context: I'm using gitlab's MlFlow server to log and register artifact. Everything that I'm logging/saving is correctly reflecting in gitlab, however, for some reason I'm getting the mflow.exceptions.RestException. I'm using torchrun to start the trainer class and there are hook registered which are executed before and after the each epoch/iteration. All-in-all it's not bothering the workflow but it'll be nice to know why I'm getting this error so I can handle it better rather than ignoring it.

Error Image here

You can also see the code for logging in one of the hook class below:

Code that integrates MlFlow into training