MLFlow logger breaks training
I'm running into two issues when I try and train with mflow installed.

The main one is transformers logs every single parameter and mlflow has restrictions on parameter size. So when I tried to train a `AutoModelForSequenceClassification` using the code below

```
num_labels=25
config = AutoConfig.from_pretrained(
    './models/roberta',
    num_labels=num_labels)
model = AutoModelForSequenceClassification.from_pretrained(
    './models/roberta', 
    config=config)
```

An exception is thrown by the mlflow/utils/validation.py - specifically that the length of the value of parameter with key "id2label" is > 250 (this is a character limit so it's trying to send the dict as a string)

The second issue is if I resume training after this error I get a further mlflow exception saying that the run is still in progress - I have to manually call

`mlflow.end_run()`

I'm not sure how to work around this (for now I've uninstalled mlflow). Can I tell the trainer not to log certain parameters?

This issue has been automatically marked as stale and been closed because it has not had recent activity. Thank you for your contributions.

If you think this still needs to be addressed please comment on this thread.