Skip to content

Commit

Permalink
Fixed FID docstring and use latest pytorch_fid package (#3222)
Browse files Browse the repository at this point in the history
  • Loading branch information
vfdev-5 committed Mar 29, 2024
1 parent b8fc451 commit 99a5be6
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
7 changes: 4 additions & 3 deletions ignite/metrics/gan/fid.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion requirements-dev.txt
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ dill
setuptools
# Test contrib dependencies
scipy
pytorch_fid==0.1.1
pytorch_fid
tqdm
scikit-learn
matplotlib
Expand Down

0 comments on commit 99a5be6

Please sign in to comment.