Trainer.predict return predictions=None

Beginners
1

Hi,
I trained a Trainer and I want to apply predict.
I have a test dataset:

Dataset({
features: [‘id’, ‘query’, ‘answer’, ‘choices’, ‘gold’],
num_rows: 1422
})

But I get as the output of trainer.predict(dataset[‘train’]):

PredictionOutput(predictions=None, label_ids=None, metrics={‘test_runtime’: 0.0074, ‘test_samples_per_second’: 0.0, ‘test_steps_per_second’: 0.0})

Any idea why?

2

Having the same bug. Did you find the solution/ reason?