nv-tlabs/LION

Where is the definition of the class_embedding function?

Closed this issue · 2 comments

Where is the definition of the class_embedding function?

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.

Thank you!