[train] Add support for multiple validation dataloaders

During validation logging, lightning seems to append "/dataloader_idx_n"
to the key we define if multiple dataloaders are used.