Fix issue where batch_invstd not being set
batch_invstd can be falsely assumed to be set, even though it is None
since hasattr will not return false in this case
BatchNorm2D a reshape will be attempted then, which causes an exception
@@ -34,7 +34,7 @@ class BatchNorm2D:
else:
batch_mean, batch_var = self.running_mean, self.running_var
# NOTE: this can be precomputed for static inference. if you manually update running_var, you have to reset this
- if not hasattr(self, "batch_invstd"):
+ if not hasattr(self, "batch_invstd") or not self.batch_invstd:
self.batch_invstd = batch_var.add(self.eps)**-0.5
batch_invstd = self.batch_invstd
