File tree 2 files changed +5
-4
lines changed
2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -82,12 +82,13 @@ class FID(_BaseInceptionMetric):
82
82
83
83
Args:
84
84
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 .
86
86
feature_extractor: a torch Module for extracting the features from the input data.
87
87
It returns a tensor of shape (batch_size, num_features).
88
88
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.
91
92
Please note that the model will be implicitly converted to device mentioned in the ``device``
92
93
argument.
93
94
output_transform: a callable that is used to transform the
Original file line number Diff line number Diff line change 7
7
setuptools
8
8
# Test contrib dependencies
9
9
scipy
10
- pytorch_fid == 0.1.1
10
+ pytorch_fid
11
11
tqdm
12
12
scikit-learn
13
13
matplotlib
You can’t perform that action at this time.
0 commit comments