Index

B

batch_size (torch_fidelity.calculate_metrics parameter)

C

cache (torch_fidelity.calculate_metrics parameter) cache_root (torch_fidelity.calculate_metrics parameter) calculate_metrics() (in module torch_fidelity)

cls (torch_fidelity.register_feature_extractor parameter) (torch_fidelity.register_sample_similarity parameter) cuda (torch_fidelity.calculate_metrics parameter) (torch_fidelity.GenerativeModelModuleWrapper parameter)

D

datasets_download (torch_fidelity.calculate_metrics parameter)

datasets_root (torch_fidelity.calculate_metrics parameter)

F

feature_extractor (torch_fidelity.calculate_metrics parameter) feature_extractor_weights_path (torch_fidelity.calculate_metrics parameter) (torch_fidelity.FeatureExtractorInceptionV3 parameter) feature_layer_fid (torch_fidelity.calculate_metrics parameter) feature_layer_isc (torch_fidelity.calculate_metrics parameter) feature_layer_kid (torch_fidelity.calculate_metrics parameter) FeatureExtractorBase (class in torch_fidelity) FeatureExtractorInceptionV3 (class in torch_fidelity)

features_list (torch_fidelity.FeatureExtractorBase parameter) (torch_fidelity.FeatureExtractorInceptionV3 parameter) fid (torch_fidelity.calculate_metrics parameter) fn_create (torch_fidelity.register_dataset parameter) fn_generate (torch_fidelity.register_noise_source parameter) fn_interpolate (torch_fidelity.register_interpolation parameter) forward() (torch_fidelity.FeatureExtractorBase method) (torch_fidelity.SampleSimilarityBase method)

G

GenerativeModelBase (class in torch_fidelity) GenerativeModelModuleWrapper (class in torch_fidelity)

GenerativeModelONNX (class in torch_fidelity) get_provided_features_list() (torch_fidelity.FeatureExtractorBase static method)

I

input1 (torch_fidelity.calculate_metrics parameter) input1_cache_name (torch_fidelity.calculate_metrics parameter) input1_model_num_classes (torch_fidelity.calculate_metrics parameter) input1_model_num_samples (torch_fidelity.calculate_metrics parameter) input1_model_z_size (torch_fidelity.calculate_metrics parameter) input1_model_z_type (torch_fidelity.calculate_metrics parameter) input2 (torch_fidelity.calculate_metrics parameter)

input2_cache_name (torch_fidelity.calculate_metrics parameter) input2_model_num_classes (torch_fidelity.calculate_metrics parameter) input2_model_num_samples (torch_fidelity.calculate_metrics parameter) input2_model_z_size (torch_fidelity.calculate_metrics parameter) input2_model_z_type (torch_fidelity.calculate_metrics parameter) isc (torch_fidelity.calculate_metrics parameter) isc_splits (torch_fidelity.calculate_metrics parameter)

K

kid (torch_fidelity.calculate_metrics parameter) kid_coef0 (torch_fidelity.calculate_metrics parameter) kid_degree (torch_fidelity.calculate_metrics parameter)

kid_gamma (torch_fidelity.calculate_metrics parameter) kid_subset_size (torch_fidelity.calculate_metrics parameter) kid_subsets (torch_fidelity.calculate_metrics parameter)

M

make_copy (torch_fidelity.GenerativeModelModuleWrapper parameter) make_eval (torch_fidelity.GenerativeModelModuleWrapper parameter)

module torch_fidelity module (torch_fidelity.GenerativeModelModuleWrapper parameter)

N

name (torch_fidelity.FeatureExtractorBase parameter) (torch_fidelity.FeatureExtractorInceptionV3 parameter) (torch_fidelity.register_dataset parameter) (torch_fidelity.register_feature_extractor parameter) (torch_fidelity.register_interpolation parameter) (torch_fidelity.register_noise_source parameter) (torch_fidelity.register_sample_similarity parameter) (torch_fidelity.SampleSimilarityBase parameter) (torch_fidelity.SampleSimilarityLPIPS parameter)

num_classes (torch_fidelity.GenerativeModelBase property) (torch_fidelity.GenerativeModelModuleWrapper parameter) (torch_fidelity.GenerativeModelONNX parameter)

P

path_onnx (torch_fidelity.GenerativeModelONNX parameter) ppl (torch_fidelity.calculate_metrics parameter) ppl_discard_percentile_higher (torch_fidelity.calculate_metrics parameter) ppl_discard_percentile_lower (torch_fidelity.calculate_metrics parameter) ppl_epsilon (torch_fidelity.calculate_metrics parameter)

ppl_reduction (torch_fidelity.calculate_metrics parameter) ppl_sample_similarity (torch_fidelity.calculate_metrics parameter) ppl_sample_similarity_dtype (torch_fidelity.calculate_metrics parameter) ppl_sample_similarity_resize (torch_fidelity.calculate_metrics parameter) ppl_z_interp_mode (torch_fidelity.calculate_metrics parameter)

R

register_dataset() (in module torch_fidelity) register_feature_extractor() (in module torch_fidelity) register_interpolation() (in module torch_fidelity)

register_noise_source() (in module torch_fidelity) register_sample_similarity() (in module torch_fidelity) rng_seed (torch_fidelity.calculate_metrics parameter)

S

sample_similarity_dtype (torch_fidelity.SampleSimilarityLPIPS parameter) sample_similarity_resize (torch_fidelity.SampleSimilarityLPIPS parameter) samples_ext_lossy (torch_fidelity.calculate_metrics parameter) samples_find_deep (torch_fidelity.calculate_metrics parameter)

samples_find_ext (torch_fidelity.calculate_metrics parameter) samples_shuffle (torch_fidelity.calculate_metrics parameter) SampleSimilarityBase (class in torch_fidelity) SampleSimilarityLPIPS (class in torch_fidelity) save_cpu_ram (torch_fidelity.calculate_metrics parameter)

T

torch_fidelity module

V

verbose (torch_fidelity.calculate_metrics parameter)

Z

z_size (torch_fidelity.GenerativeModelBase property) (torch_fidelity.GenerativeModelModuleWrapper parameter) (torch_fidelity.GenerativeModelONNX parameter)

z_type (torch_fidelity.GenerativeModelBase property) (torch_fidelity.GenerativeModelModuleWrapper parameter) (torch_fidelity.GenerativeModelONNX parameter)