8 lines
146 B
Python
8 lines
146 B
Python
|
"""Loss type enums."""
|
||
|
from enum import Enum
|
||
|
|
||
|
|
||
|
class LossType(str, Enum):
|
||
|
CROSS_ENTROPY = "cross_entropy"
|
||
|
BCE_WITH_LOGITS = "bce_with_logits"
|