I am trying to use the torch vision.detection.keypointrcnn model in Pytorch Lightning.
The model returns a dictionary of losses in training mode,
loss_dict = model(images, targets).
I would like the model to return the validation loss in the validation step, but I am getting predictions instead of the losses. Should I implement the loss myself?
Thanks in advance.