hpcaitech / Open-Sora

Open-Sora: Democratizing Efficient Video Production for All
https://hpcaitech.github.io/Open-Sora/
Apache License 2.0
20.1k stars 1.9k forks source link

The training problem #542

Open f1yfisher opened 1 week ago

f1yfisher commented 1 week ago

I found that the number of output channels of STDiT-v3 is 8, but in the training phase, only the former 4 channels are computed in the loss function.

JThh commented 3 hours ago

Thanks. Would you please help me pinpoint where you identified this issue? I saw this but failed to locate where channel selection happened.