Unconditional Diffusion Example without Grayscale

[codeHorasan]

How to run the Unconditional Diffusion example on custom dataset without grayscale? It returns error whenever the dataset is not grayscaled

[cabralpinto]

@codeHorasan Could you provide the code snippet in question and the error you get? The library should work for any data type but you might need to reshape it depending on the network and the noise type you choose.

[cabralpinto]

Will close due to inactivity.