We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a3d0922 commit 596aa34Copy full SHA for 596aa34
denoising_diffusion_pytorch/fid_evaluation.py
@@ -48,7 +48,10 @@ def __init__(
48
def calculate_inception_features(self, samples):
49
if self.channels == 1:
50
samples = repeat(samples, "b 1 ... -> b c ...", c=3)
51
+
52
+ self.inception_v3.eval()
53
features = self.inception_v3(samples)[0]
54
55
if features.size(2) != 1 or features.size(3) != 1:
56
features = adaptive_avg_pool2d(features, output_size=(1, 1))
57
features = rearrange(features, "... 1 1 -> ...")
denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
1
-__version__ = '1.8.4'
+__version__ = '1.8.5'
0 commit comments