diff --git a/ignite/metrics/gan/fid.py b/ignite/metrics/gan/fid.py index d3ec32707b7a..188bad5035a2 100644 --- a/ignite/metrics/gan/fid.py +++ b/ignite/metrics/gan/fid.py @@ -82,12 +82,13 @@ class FID(_BaseInceptionMetric): Args: num_features: number of features predicted by the model or the reduced feature vector of the image. - Default value is 2048. + Default value is 1000. feature_extractor: a torch Module for extracting the features from the input data. It returns a tensor of shape (batch_size, num_features). If neither ``num_features`` nor ``feature_extractor`` are defined, by default we use an ImageNet - pretrained Inception Model. If only ``num_features`` is defined but ``feature_extractor`` is not - defined, ``feature_extractor`` is assigned Identity Function. + pretrained Inception Model and use model's output logits as features. + If only ``num_features`` is defined but ``feature_extractor`` is not defined, + ``feature_extractor`` is assigned Identity Function. Please note that the model will be implicitly converted to device mentioned in the ``device`` argument. output_transform: a callable that is used to transform the diff --git a/requirements-dev.txt b/requirements-dev.txt index bd5aca1f411c..7739bd7a0575 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -7,7 +7,7 @@ dill setuptools # Test contrib dependencies scipy -pytorch_fid==0.1.1 +pytorch_fid tqdm scikit-learn matplotlib