Update functions.py

2025-03-18 09:23:09 +09:00
parent 7baf452d88
commit 6853a794c2
1 changed files with 3 additions and 1 deletions
@@ -6,6 +6,8 @@ from torch.autograd.function import once_differentiable

 # Activation names
 ACT_LEAKY_RELU = "leaky_relu"
+ACT_RELU = "relu"
+ACT_ELU = "elu"
 ACT_NONE = "none"


@@ -174,4 +176,4 @@ class InPlaceABNSync(autograd.Function):
 inplace_abn = InPlaceABN.apply
 inplace_abn_sync = InPlaceABNSync.apply

-__all__ = ["inplace_abn", "inplace_abn_sync", "ACT_LEAKY_RELU", "ACT_NONE"]
+__all__ = ["inplace_abn", "inplace_abn_sync", "ACT_LEAKY_RELU", "ACT_RELU", "ACT_ELU", "ACT_NONE"]