From 6853a794c23a8dab45cfeca03099d6c67c38755d Mon Sep 17 00:00:00 2001 From: "Jinwoo Park (Curt)" Date: Tue, 18 Mar 2025 09:23:09 +0900 Subject: [PATCH] Update functions.py --- schp/modules/functions.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/schp/modules/functions.py b/schp/modules/functions.py index 64e5a08..f403f73 100644 --- a/schp/modules/functions.py +++ b/schp/modules/functions.py @@ -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"]