Diffusion-Models-pytorch icon indicating copy to clipboard operation
Diffusion-Models-pytorch copied to clipboard
Published 2 years ago • verified publisher icon dome272

Removed hardcoded classes length

Open SuperSecureHuman opened this issue 2 years ago • 0 comments

Removed the hardcoded classes length value of 10. Replaced it with args.num_classes. Now this allows us to experiment with this model across different datasets, without errors.

The hardcoded value, throws an error when used with other class length with the embedding layer.

SuperSecureHuman avatar Sep 07 '23 09:09 SuperSecureHuman