Fix multiple dev/test datasets after restoring from checkpoint
Signed-off-by: PeganovAnton <peganoff2@mail.ru>

# What does this PR do ?

When model was restored from checkpoint, several datasets were not working for testing and validation.

**Collection**: nlp
