

TITLE
MLFlow logger breaks training

COMMENTS
1

REACTIONS
+1: 0
-1: 0
laugh: 0
hooray: 0
heart: 0
rocket: 0
eyes: 0

BODY
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?
