1from enum import Enum 

2import torch.nn.functional as F 

3 

4 

5class ContinuousLossType(Enum): 

6 L1_LOSS = 0 

7 MSE_LOSS = 1 

8 SMOOTH_L1_LOSS = 2 

9 

10 

11class BinaryLossType(Enum): 

12 CROSS_ENTROPY = 0 

13 IOU = 1 

14 DICE = 2 

15 

16 

17class CategoricalLossType(Enum): 

18 CROSS_ENTROPY = 0 

19 DICE = 1 

20 # TODO: Focal Loss 

21 # TODO Wasserstein distance (Earth mover loss) 

22