From f0d4eb9182829d26e5eda14545e3f7ca6e3c2cf0 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 7 Jun 2021 11:58:36 -0600 Subject: [PATCH] Fixor --- codes/models/classifiers/cifar_resnet_branched.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/classifiers/cifar_resnet_branched.py b/codes/models/classifiers/cifar_resnet_branched.py index 5eda08f8..e37b2586 100644 --- a/codes/models/classifiers/cifar_resnet_branched.py +++ b/codes/models/classifiers/cifar_resnet_branched.py @@ -166,7 +166,7 @@ class DropoutNorm(SwitchNorm): # Ensure that there is always at least one switch left un-dropped out fix_blank = (drop.sum(dim=1, keepdim=True) == 0).repeat(1, br) drop = drop.logical_or(fix_blank) - x = drop * x + ((not drop) * x * self.eps) + x = drop * x + ((~drop) * x * self.eps) return x