Add CrossEntropyLoss
and DiceCELoss
wrappers with support for class weights
#791
Job | Run time |
---|---|
2m 12s | |
8m 29s | |
1m 39s | |
4s | |
12m 24s |