sail-sg / MDT

Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
Apache License 2.0
500 stars 35 forks source link

training problem #38

Open ZH3nGsQ opened 5 months ago

ZH3nGsQ commented 5 months ago

hello! when I am training with one gpu. I encountered with a problem: image and I set the script like this image I don't know which leaf Variable that requires grad is being used in an in-place operation.

gasvn commented 4 months ago

Can you show me how to reproduce the error?

ZH3nGsQ commented 4 months ago

image I just run the code block and this error occurred.