How to make a custom loss function (PyTorch)

Part 1 (2018)
28

Some pytorch loss functions allow class weights to be passed in. Have you ruled that approach out? Eg nn.CrossEntropyLoss(weight=class_weights)