Do you care more about a specific class or is each per class accuracy of equal importance?
Based on the class counts, you could try weight=torch.tensor([4, 0.23, 0.25]).
Note that the weights do not have to sum to one. You could also try your weights first.
Did you observe the confusion matrix? Could you post it using the validation dataset?