Skip to content

Commit 596aa34

Browse files
committed
make sure inception v3 is in eval mode for fid, thanks to @suheey
1 parent a3d0922 commit 596aa34

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

denoising_diffusion_pytorch/fid_evaluation.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,10 @@ def __init__(
4848
def calculate_inception_features(self, samples):
4949
if self.channels == 1:
5050
samples = repeat(samples, "b 1 ... -> b c ...", c=3)
51+
52+
self.inception_v3.eval()
5153
features = self.inception_v3(samples)[0]
54+
5255
if features.size(2) != 1 or features.size(3) != 1:
5356
features = adaptive_avg_pool2d(features, output_size=(1, 1))
5457
features = rearrange(features, "... 1 1 -> ...")
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = '1.8.4'
1+
__version__ = '1.8.5'

0 commit comments

Comments
 (0)