From 765fb01c09f978bdbc13fc7c32c28e8c6ff7e085 Mon Sep 17 00:00:00 2001 From: "Jinwoo Park (Curt)" Date: Tue, 18 Mar 2025 09:33:50 +0900 Subject: [PATCH] Update functions.py --- schp/modules/functions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/schp/modules/functions.py b/schp/modules/functions.py index 194e877..d41afe3 100644 --- a/schp/modules/functions.py +++ b/schp/modules/functions.py @@ -14,7 +14,7 @@ ACT_NONE = "none" class InPlaceABN(autograd.Function): @staticmethod def forward(ctx, x, weight, bias, running_mean, running_var, - training=True, momentum=0.1, eps=1e-05, activation='leaky_relu', slope=0.01): + training=True, momentum=0.1, eps=1e-05, activation="leaky_relu", slope=0.01): ctx.training = training ctx.momentum = momentum ctx.eps = eps @@ -84,7 +84,7 @@ class InPlaceABN(autograd.Function): class InPlaceABNSync(autograd.Function): @classmethod def forward(cls, ctx, x, weight, bias, running_mean, running_var, - training=True, momentum=0.1, eps=1e-05, activation='leaky_relu', slope=0.01, equal_batches=True): + training=True, momentum=0.1, eps=1e-05, activation="leaky_relu", slope=0.01, equal_batches=True): ctx.training = training ctx.momentum = momentum ctx.eps = eps