Index

_

__call__() (pe.callback.callback.Callback method) (pe.callback.common.compute_fid.ComputeFID method) (pe.callback.common.save_checkpoints.SaveCheckpoints method) (pe.callback.image.sample_images.SampleImages method) (pe.callback.image.save_all_images.SaveAllImages method) __init__() (pe.api.image.improved_diffusion_api.ImprovedDiffusion method) (pe.api.image.improved_diffusion_api.ImprovedDiffusion270M method) (pe.api.image.stable_diffusion_api.StableDiffusion method) (pe.callback.common.compute_fid.ComputeFID method) (pe.callback.common.save_checkpoints.SaveCheckpoints method) (pe.callback.image.sample_images.SampleImages method) (pe.callback.image.save_all_images.SaveAllImages method) (pe.data.data.Data method) (pe.data.image.camelyon17.Camelyon17 method) (pe.data.image.cat.Cat method) (pe.data.image.cifar10.Cifar10 method) (pe.embedding.image.inception.Inception method) (pe.histogram.nearest_neighbors.NearestNeighbors method) (pe.logger.csv_print.CSVPrint method) (pe.logger.image_file.ImageFile method) (pe.logger.log_print.LogPrint method) (pe.logger.matplotlib_pdf.MatplotlibPDF method) (pe.metric_item.ImageListMetricItem method) (pe.metric_item.MetricItem method) (pe.population.pe_population.PEPopulation method) (pe.runner.pe.PE method)

_clean_up_loggers() (pe.runner.pe.PE method) _clear_logs() (pe.logger.csv_print.CSVPrint method) _compute_lookahead_embedding() (pe.histogram.nearest_neighbors.NearestNeighbors method) _download() (pe.data.image.cat.Cat method) _flush() (pe.logger.csv_print.CSVPrint method) _get_checkpoint_path() (pe.callback.common.save_checkpoints.SaveCheckpoints method) _get_image_path() (pe.logger.image_file.ImageFile method) _get_log_path() (pe.logger.csv_print.CSVPrint method) _get_num_samples_per_label_id() (pe.runner.pe.PE method) _get_pdf_path() (pe.logger.matplotlib_pdf.MatplotlibPDF method) _list_image_files_recursively() (in module pe.data.image.image) _log_float() (pe.logger.csv_print.CSVPrint method) _log_image() (pe.logger.image_file.ImageFile method) _log_image_list() (pe.logger.image_file.ImageFile method) _log_lookahead() (pe.histogram.nearest_neighbors.NearestNeighbors method) _log_metrics() (pe.runner.pe.PE method) _log_voting_details() (pe.histogram.nearest_neighbors.NearestNeighbors method) _post_process_histogram() (pe.population.pe_population.PEPopulation method) _read_data() (pe.data.image.cat.Cat method) _save_image() (pe.callback.image.save_all_images.SaveAllImages method) _select_data() (pe.population.pe_population.PEPopulation method)

A

add_noise() (pe.dp.dp.DP method) (pe.dp.gaussian.Gaussian method)

API (class in pe.api.api)

C

Callback (class in pe.callback.callback) Camelyon17 (class in pe.data.image.camelyon17) Cat (class in pe.data.image.cat) CHECKPOINT_URL (pe.api.image.improved_diffusion_api.ImprovedDiffusion270M attribute) Cifar10 (class in pe.data.image.cifar10) CLEAN_HISTOGRAM_COLUMN_NAME (in module pe.constant.data) clean_up() (pe.logger.csv_print.CSVPrint method) (pe.logger.logger.Logger method) (pe.metric_item.MatplotlibMetricItem method) (pe.metric_item.MetricItem method) column_name (pe.embedding.embedding.Embedding property)

compute_embedding() (pe.embedding.embedding.Embedding method) (pe.embedding.image.inception.Inception method) compute_epsilon() (in module pe.dp.gaussian) compute_histogram() (pe.histogram.histogram.Histogram method) (pe.histogram.nearest_neighbors.NearestNeighbors method) ComputeFID (class in pe.callback.common.compute_fid) concat() (pe.data.data.Data class method) ConstantList (class in pe.api.util) create_gaussian_diffusion() (in module pe.api.image.improved_diffusion_lib.gaussian_diffusion) create_model() (in module pe.api.image.improved_diffusion_lib.unet) CSVPrint (class in pe.logger.csv_print)

D

Data (class in pe.data.data) ddim_reverse_sample() (pe.api.image.improved_diffusion_lib.gaussian_diffusion.SkippedSpacedDiffusion method) ddim_sample() (pe.api.image.improved_diffusion_lib.gaussian_diffusion.SkippedSpacedDiffusion method) ddim_sample_loop() (pe.api.image.improved_diffusion_lib.gaussian_diffusion.SkippedSpacedDiffusion method)

ddim_sample_loop_progressive() (pe.api.image.improved_diffusion_lib.gaussian_diffusion.SkippedSpacedDiffusion method) delta_Gaussian() (in module pe.dp.gaussian) download() (in module pe.util.download) DP (class in pe.dp.dp) DP_HISTOGRAM_COLUMN_NAME (in module pe.constant.data)

E

Embedding (class in pe.embedding.embedding) EMBEDDING_COLUMN_NAME (in module pe.constant.data)

eps_Gaussian() (in module pe.dp.gaussian) execution_logger (in module pe.logging)

F

filter_label_id() (pe.data.data.Data method) filter_uncomputed_rows() (pe.embedding.embedding.Embedding method) FloatListMetricItem (class in pe.metric_item)

FloatMetricItem (class in pe.metric_item) forward() (pe.api.image.improved_diffusion_api.Sampler method) FP32UNetModel (class in pe.api.image.improved_diffusion_lib.unet) FROM_LAST_FLAG_COLUMN_NAME (in module pe.constant.data)

G

Gaussian (class in pe.dp.gaussian)

get_noise_multiplier() (in module pe.dp.gaussian)

H

Histogram (class in pe.histogram.histogram)

HISTOGRAM_NEAREST_NEIGHBORS_VOTING_IDS_COLUMN_NAME (in module pe.constant.data)

I

IMAGE_DATA_COLUMN_NAME (in module pe.constant.data) IMAGE_MODEL_LABEL_COLUMN_NAME (in module pe.constant.data) IMAGE_PROMPT_COLUMN_NAME (in module pe.constant.data) ImageDataset (class in pe.data.image.image) ImageFile (class in pe.logger.image_file) ImageListMetricItem (class in pe.metric_item)

ImageMetricItem (class in pe.metric_item) ImprovedDiffusion (class in pe.api.image.improved_diffusion_api) ImprovedDiffusion270M (class in pe.api.image.improved_diffusion_api) Inception (class in pe.embedding.image.inception) initial() (pe.population.pe_population.PEPopulation method) (pe.population.population.Population method) inner_dtype (pe.api.image.improved_diffusion_lib.unet.FP32UNetModel property)

L

LABEL_ID_COLUMN_NAME (in module pe.constant.data) load_checkpoint() (pe.data.data.Data method) (pe.runner.pe.PE method) load_image_folder() (in module pe.data.image.image) log() (pe.logger.csv_print.CSVPrint method) (pe.logger.image_file.ImageFile method) (pe.logger.log_print.LogPrint method) (pe.logger.logger.Logger method) (pe.logger.matplotlib_pdf.MatplotlibPDF method)

Logger (class in pe.logger.logger) LogPrint (class in pe.logger.log_print) LOOKAHEAD_EMBEDDING_COLUMN_NAME (in module pe.constant.data)

M

MatplotlibMetricItem (class in pe.metric_item) MatplotlibPDF (class in pe.logger.matplotlib_pdf) merge() (pe.data.data.Data method) merge_computed_rows() (pe.embedding.embedding.Embedding method) metric_scope (class in pe.metric_item) MetricItem (class in pe.metric_item) module pe pe.api pe.api.api pe.api.image pe.api.image.improved_diffusion_api pe.api.image.improved_diffusion_lib pe.api.image.improved_diffusion_lib.gaussian_diffusion pe.api.image.improved_diffusion_lib.unet pe.api.image.stable_diffusion_api pe.api.util pe.callback pe.callback.callback pe.callback.common pe.callback.common.compute_fid pe.callback.common.save_checkpoints pe.callback.image pe.callback.image.sample_images pe.callback.image.save_all_images pe.constant pe.constant.data pe.data pe.data.data pe.data.image pe.data.image.camelyon17 pe.data.image.cat pe.data.image.cifar10 pe.data.image.image pe.dp pe.dp.dp pe.dp.gaussian pe.embedding pe.embedding.embedding pe.embedding.image pe.embedding.image.inception pe.histogram pe.histogram.histogram pe.histogram.nearest_neighbor_backend pe.histogram.nearest_neighbor_backend.auto pe.histogram.nearest_neighbor_backend.faiss pe.histogram.nearest_neighbor_backend.sklearn pe.histogram.nearest_neighbors pe.logger pe.logger.csv_print pe.logger.image_file pe.logger.log_print pe.logger.logger pe.logger.matplotlib_pdf pe.logging pe.metric_item pe.population pe.population.pe_population pe.population.population pe.runner pe.runner.pe pe.util pe.util.download

N

name (pe.metric_item.MetricItem property) NearestNeighbors (class in pe.histogram.nearest_neighbors)

next() (pe.population.pe_population.PEPopulation method) (pe.population.population.Population method) num_images_per_row (pe.metric_item.ImageListMetricItem property)

P

p_sample_loop() (pe.api.image.improved_diffusion_lib.gaussian_diffusion.SkippedSpacedDiffusion method) p_sample_loop_progressive() (pe.api.image.improved_diffusion_lib.gaussian_diffusion.SkippedSpacedDiffusion method) PARENT_SYN_DATA_INDEX_COLUMN_NAME (in module pe.constant.data) pe module PE (class in pe.runner.pe) pe.api module pe.api.api module pe.api.image module pe.api.image.improved_diffusion_api module pe.api.image.improved_diffusion_lib module pe.api.image.improved_diffusion_lib.gaussian_diffusion module pe.api.image.improved_diffusion_lib.unet module pe.api.image.stable_diffusion_api module pe.api.util module pe.callback module pe.callback.callback module pe.callback.common module pe.callback.common.compute_fid module pe.callback.common.save_checkpoints module pe.callback.image module pe.callback.image.sample_images module pe.callback.image.save_all_images module pe.constant module pe.constant.data module pe.data module pe.data.data module pe.data.image module pe.data.image.camelyon17 module pe.data.image.cat module pe.data.image.cifar10 module pe.data.image.image module pe.dp module

pe.dp.dp module pe.dp.gaussian module pe.embedding module pe.embedding.embedding module pe.embedding.image module pe.embedding.image.inception module pe.histogram module pe.histogram.histogram module pe.histogram.nearest_neighbor_backend module pe.histogram.nearest_neighbor_backend.auto module pe.histogram.nearest_neighbor_backend.faiss module pe.histogram.nearest_neighbor_backend.sklearn module pe.histogram.nearest_neighbors module pe.logger module pe.logger.csv_print module pe.logger.image_file module pe.logger.log_print module pe.logger.logger module pe.logger.matplotlib_pdf module pe.logging module pe.metric_item module pe.population module pe.population.pe_population module pe.population.population module pe.runner module pe.runner.pe module pe.util module pe.util.download module PEPopulation (class in pe.population.pe_population) Population (class in pe.population.population) POST_PROCESSED_DP_HISTOGRAM_COLUMN_NAME (in module pe.constant.data)

R

random_api() (pe.api.api.API method) (pe.api.image.improved_diffusion_api.ImprovedDiffusion method) (pe.api.image.stable_diffusion_api.StableDiffusion method)

random_truncate() (pe.data.data.Data method) run() (pe.runner.pe.PE method)

S

sample() (in module pe.api.image.improved_diffusion_api) SampleImages (class in pe.callback.image.sample_images) Sampler (class in pe.api.image.improved_diffusion_api) save_checkpoint() (pe.data.data.Data method) SaveAllImages (class in pe.callback.image.save_all_images) SaveCheckpoints (class in pe.callback.common.save_checkpoints) search() (in module pe.histogram.nearest_neighbor_backend.auto) (in module pe.histogram.nearest_neighbor_backend.faiss) (in module pe.histogram.nearest_neighbor_backend.sklearn)

set_epsilon_and_delta() (pe.dp.dp.DP method) (pe.dp.gaussian.Gaussian method) set_label_id() (pe.data.data.Data method) setup_logging() (in module pe.logging) SkippedSpacedDiffusion (class in pe.api.image.improved_diffusion_lib.gaussian_diffusion) StableDiffusion (class in pe.api.image.stable_diffusion_api)

T

to_uint8() (in module pe.embedding.image.inception) training (pe.api.image.improved_diffusion_api.Sampler attribute) (pe.api.image.improved_diffusion_lib.unet.FP32UNetModel attribute)

truncate() (pe.data.data.Data method)

U

URL (pe.data.image.cat.Cat attribute)

V

value (pe.metric_item.MetricItem property) variation_api() (pe.api.api.API method) (pe.api.image.improved_diffusion_api.ImprovedDiffusion method) (pe.api.image.stable_diffusion_api.StableDiffusion method)