Skip to content

Commit 99a5be6

Browse files
authored
Fixed FID docstring and use latest pytorch_fid package (#3222)
1 parent b8fc451 commit 99a5be6

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

ignite/metrics/gan/fid.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,12 +82,13 @@ class FID(_BaseInceptionMetric):
8282
8383
Args:
8484
num_features: number of features predicted by the model or the reduced feature vector of the image.
85-
Default value is 2048.
85+
Default value is 1000.
8686
feature_extractor: a torch Module for extracting the features from the input data.
8787
It returns a tensor of shape (batch_size, num_features).
8888
If neither ``num_features`` nor ``feature_extractor`` are defined, by default we use an ImageNet
89-
pretrained Inception Model. If only ``num_features`` is defined but ``feature_extractor`` is not
90-
defined, ``feature_extractor`` is assigned Identity Function.
89+
pretrained Inception Model and use model's output logits as features.
90+
If only ``num_features`` is defined but ``feature_extractor`` is not defined,
91+
``feature_extractor`` is assigned Identity Function.
9192
Please note that the model will be implicitly converted to device mentioned in the ``device``
9293
argument.
9394
output_transform: a callable that is used to transform the

requirements-dev.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ dill
77
setuptools
88
# Test contrib dependencies
99
scipy
10-
pytorch_fid==0.1.1
10+
pytorch_fid
1111
tqdm
1212
scikit-learn
1313
matplotlib

0 commit comments

Comments
 (0)