dSprites dataset

[MoeinSorkhei]

Hello, I am unable to produce similar results to yours on dSprites dataset. You have mentioned the that the dsprites_loc and dsprites_orientation tasks both have 16 classes, while in the original repository it is mentioned they have 32 and 40 classes respectively.

Could you clarify how you select classes?
I did not fined any info about this in the code.