FID score in validation_epoch_end #6175
Unanswered
rubencart
asked this question in
DDP / multi-GPU / multi-node
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I want to compute the FID score for two lists of images stored on disk in different paths. I call
calculate_fid_given_paths
from pytorch-fid in myvalidation_epoch_end
method.For DDP, how do I make sure the fid isn't calculated twice? Would wrapping the contents of
validation_epoch_end
in aif self.local_rank == 0:
make sense? PL complains with aMisconfigurationException
though, because I have a checkpoint callback watching the computed fid score, and in this case the value is only logged on the main process. How do I avoid computing the same score twice correctly?Beta Was this translation helpful? Give feedback.
All reactions