You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
7 lines
265 B
7 lines
265 B
1 month ago
|
from examples.mlperf.metrics import dice_score
|
||
|
|
|
||
|
|
def dice_ce_loss(pred, tgt):
|
||
|
|
ce = pred.permute(0, 2, 3, 4, 1).sparse_categorical_crossentropy(tgt.squeeze(1))
|
||
|
|
dice = (1.0 - dice_score(pred, tgt, argmax=False, to_one_hot_x=False)).mean()
|
||
|
|
return (dice + ce) / 2
|