feat: Add support for passing weight to the classification loss functions (#260)
* add support for passing weight to the loss functions * adds test for weights and fixes issue with state dict
M
Marcus McCurdy committed
4867cb86e955e462fe3659b01903f69bbc88180b
Parent: 06a478c
Committed by GitHub <noreply@github.com>
on 6/6/2025, 2:38:06 PM