XiangLi1999 / Diffusion-LM

Diffusion-LM
Apache License 2.0
1.02k stars 133 forks source link

Why model.model.module instead of model.model? #44

Closed smiles724 closed 1 year ago

smiles724 commented 1 year ago

Hi, may I ask a question about DDP?

I notice that you use model.model.module to get access some customized attributes like get_embedding(). Nevertheless, as far as I know, the right way ought to be model.module.

Interestingly, model.model.module works well in your original version.

image image