sail-sg / MDT

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

Weights during training #30

Closed liwenyang-911 closed 5 months ago

liwenyang-911 commented 5 months ago

Why do I keep putting out 3 weights when I'm training? If the model... opt... ema...

image

How to use the weights obtained from training for inference, in other words, which weight should be used for inference