You could modify the weights of the criterion like this:
criterion.weight = torch.tensor([0.3, 0.3, 0.3, 0.1, 0.1, 0.1], dtype=torch.float32)

So I guess you could have a if condition on the target value and change the weight attribute as you want. Is this what you want?