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!