Fixor

2021-06-07 11:58:36 -06:00 · 2021-06-07 11:58:36 -06:00 · f0d4eb9182
commit f0d4eb9182
parent c456a60466
1 changed files with 1 additions and 1 deletions
--- 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