remove check for linear

Author: stephantul

model2vec/train/classifier.py

@@ -84,9 +84,8 @@ def construct_head(self) -> nn.Sequential:
         if linear_modules:
             *initial, last = linear_modules
             for module in initial:
-                if isinstance(module, nn.Linear):
-                    nn.init.kaiming_uniform_(module.weight, nonlinearity="relu")
-                    nn.init.zeros_(module.bias)
+                nn.init.kaiming_uniform_(module.weight, nonlinearity="relu")
+                nn.init.zeros_(module.bias)
             # Final layer does not kaiming
             nn.init.xavier_uniform_(last.weight)
             nn.init.zeros_(last.bias)