Technical University of Denmark
Browse

VDMv-32 trained on ImageNet32

Download (5.05 GB)
dataset
posted on 2024-04-22, 10:00 authored by Beatrix Miranda Ginn NielsenBeatrix Miranda Ginn Nielsen

A variational diffusion model (VDM, https://arxiv.org/abs/2107.00630) trained for the article "DiffEnc: Variational Diffusion with a Learned Encoder" (https://arxiv.org/abs/2310.19789).

Model uses v-parametrization for the loss. The diffusion model is of size 32. That is, the diffusion model uses 32 "down-blocks" in the U-net. See details in article.

Model was trained on ImageNet32 for 1.5 million steps.

Random seeds: 1, 2, 13


Funding

Danish Pioneer Centre for AI, DNRF grant number P1

History

ORCID for corresponding depositor

Usage metrics

    DTU Compute

    Categories

    Licence

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC