Standard names for loss type
| 19 | |
| 20 | |
| 21 | class LossType(Type): |
| 22 | """Standard names for loss type |
| 23 | """ |
| 24 | SOFTMAX_CROSS_ENTROPY = "SoftmaxCrossEntropy" |
| 25 | SOFTMAX_FOCAL_CROSS_ENTROPY = "SoftmaxFocalCrossEntropy" |
| 26 | SIGMOID_FOCAL_CROSS_ENTROPY = "SigmoidFocalCrossEntropy" |
| 27 | BCE_WITH_LOGITS = "BCEWithLogitsLoss" |
| 28 | |
| 29 | @classmethod |
| 30 | def str(cls): |
| 31 | return ",".join([cls.SOFTMAX_CROSS_ENTROPY, |
| 32 | cls.SOFTMAX_FOCAL_CROSS_ENTROPY, |
| 33 | cls.SIGMOID_FOCAL_CROSS_ENTROPY, |
| 34 | cls.BCE_WITH_LOGITS]) |
| 35 | |
| 36 | |
| 37 | class ActivationType(Type): |
nothing calls this directly
no outgoing calls
no test coverage detected