Open f1yfisher opened 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.
Thanks. Would you please help me pinpoint where you identified this issue? I saw this but failed to locate where channel selection happened.
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.