Validation loss is none while training using pytorch training loop
Hi! I am fine-tuning aMask2FormerForUniversalSegmentationmodel from a pre-trained checkpoint using a custom pytorch training loop. I am able to see the loss during the train step, i.e. whenmodel.train()is set. But the loss isNonein the validation step, i.e. whenmodel.eval()is set and withintorch.no_grad()ortorch.inference_mode()context.What am I missing? Any ideas on what I could do to get the validation loss?