search

jolibrain / joliGEN

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications
https://www.joligen.com
Other
238 stars 31 forks source link

feat: self supervised labeled cls dataset #597

Open pierrot-lc opened 8 months ago

pierrot-lc commented 8 months ago

This new dataset allows us to make training where the model is conditionned on the class of the input only.

beniz commented 7 months ago

This is breaking the cm tests: 

E                           ValueError: test_batch_size: Bad type (<class 'NoneType'>, should be <class 'int'>)