Torchebm > Utils > Eqm_utils

Contents

Classes

• WandbLogger - Simple wandb logging wrapper.

Functions

• center_crop_arr() - Center crop and resize image to target size.
• create_npz_from_sample_folder() - Build .npz file from folder of PNG samples.
• get_vae_encode_decode() - Get VAE encoder/decoder functions for latent space operations.
• load_checkpoint() - Load training checkpoint.
• parse_transport_args() - Add transport-related arguments to parser.
• requires_grad() - Set requires_grad flag for all model parameters.
• save_checkpoint() - Save training checkpoint.
• update_ema() - Update EMA model parameters.

API Reference

torchebm.utils.eqm_utils

Utilities for EqM training and sampling.