Do we need to check self.trainer.sanity_checking
?
#16266
adamjstewart
started this conversation in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Quick question. The docs for several
LightningModule
instance methods mention checkingself.trainer.{training|validating|testing|predicting}
to apply different logic for different stages. We're applying data augmentation for each stage inon_after_batch_transfer
like so:(augmentation attributes are defined in
__init__
)Do we also need to check
self.trainer.sanity_checking
to make sure the augmentations are applied during sanity checks? Or doesself.traininer.validating
get set to True when sanity checking?@ashnair1
Beta Was this translation helpful? Give feedback.
All reactions