Qualcomm-AI-research / InverseForm

BSD 3-Clause Clear License
147 stars 22 forks source link

Question about the code #12

Open Ianresearch opened 3 years ago

Ianresearch commented 3 years ago

Hi, I have a question about the loss code. In the function of ImageBasedCrossEntropyLoss2d, the result of nll_loss.weight is no difference no matter the batch_weights is setted for True or False. Is this the original purpose that the Ith target(target[i]) should be as the input to the weights calculate when the batch_weights is setted for False?