search

nv-tlabs / LION

Latent Point Diffusion Models for 3D Shape Generation
Other
735 stars 57 forks source link

Where is the definition of the class_embedding function? #60

Closed OswaldoBornemann closed 7 months ago

OswaldoBornemann commented 8 months ago

Where is the definition of the class_embedding function?

ZENGXH commented 7 months ago

by default this function is not used, (cond_on_cat=False) if you want to add support for class embedding you could create a nn Embedding layer for it. the class conditioning support may not be completed in the code base, so there might be other places also requiring extra coding.

OswaldoBornemann commented 7 months ago

Thank you!