Some problems with graio loading training weights

PixArt-alpha / PixArt-sigma

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation

https://pixart-alpha.github.io/PixArt-sigma-project/

GNU Affero General Public License v3.0

1.63k stars 77 forks source link

Some problems with graio loading training weights #59

Closed kelisiya closed 5 months ago

kelisiya commented 5 months ago

When I trained for several rounds of epochs and wanted to use Gradio for testing, I found that epoch_1 and epoch_2 were fine, but epoch_3 "WARNING - Unexpected keys: ['blocks.14.attn.sr.weight',", and then the inference crashed, and I want to know what happened

lawrence-cj commented 5 months ago

train script, logs, test scipts, config file. Without these I can't help you.

kelisiya commented 5 months ago

train script, logs, test scipts, config file. Without these I can't help you.

I found the problem, sorry, when I wrongly loaded the weight training the saved results and the configuration file did not match