modelzoo.vision.pytorch.dit.sample_generator_dit#
Functions
Classes
Class for DiT model sample generation :param model_ckpt_path: Path to pretrained diffusion model checkpoint :type model_ckpt_path: str :param vae_ckpt_path: Path to pretrained VAE model checkpoint :type vae_ckpt_path: str :param params: Path to yaml containing model params :type params: str :param sample_dir: Path to folder where generated images and npz file to be stored :type sample_dir: str :param seed: Seed for random generation process :type seed: int :param num_fid_samples: Number of images to be generated :type num_fid_samples: int :param per_gpu_batch_size: Per gpu batch size, command line input overrides that in yaml if provided. :type per_gpu_batch_size: int. |