MamaPearl
/

nula-cifar10-robust-v0

Image Classification

computer-vision

adversarial-robustness

Model card Files Files and versions

MamaPearl commited on Apr 1

Commit

d9a85a3

·

verified ·

1 Parent(s): 39f47a1

Update modeling_nula.py

Files changed (1) hide show

modeling_nula.py +1 -1

modeling_nula.py CHANGED Viewed

@@ -144,7 +144,7 @@ class NulaForImageClassification(PreTrainedModel):
         elif isinstance(module, nn.Linear):
             nn.init.xavier_uniform_(module.weight)
             nn.init.zeros_(module.bias)
-        if isinstance(module, NULAConvBlock) and hasattr(module, "bn2"):
             nn.init.zeros_(module.bn2.weight)
     def forward(self, pixel_values=None, labels=None, return_dict=None):

         elif isinstance(module, nn.Linear):
             nn.init.xavier_uniform_(module.weight)
             nn.init.zeros_(module.bias)
+        if isinstance(module, NulaConvBlock) and hasattr(module, "bn2"):
             nn.init.zeros_(module.bn2.weight)
     def forward(self, pixel_values=None, labels=None, return_dict=None):