Siddharth Shrivastava
Results
11
comments of
Siddharth Shrivastava
Hi @vadimkantorov , As far as I know, this codebase is similar to implementation of "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions", Reference: https://github.com/ehoogeboom/multinomial_diffusion/blob/9d907a60536ad793efd6d2a6067b3c3d6ba9fce7/segmentation_diffusion/layers/layers.py#L190 Hope this helps!