Hello,
In my model , i want to backpropogate loss of specific classes only. Say if I have total 5 classes for classification. Once classifier outputs the value, in the loss computation I want to take only selected 3 classes and backpropogate the loss, others should be masked. Is there a way to achieve this in pytorch.
Thanks