I have 3 type of datasets (Let’s say all of them contain labelled images)
- special frames that I want to look at in Tensorboard (this is a third / different dataset).
I want every epoch to loop over the special frames in eval mode.
How can I achieve this using Pytorch lightning blocks?
Thanks in advance!
(Every training step I run over the training dataset, and every validation step I run over the validation dataset)