Check if the layer is part of classifier based on common naming patterns
(name: str)
| 21 | Dictionary containing parameter counts in K (1000s) |
| 22 | """ |
| 23 | def is_classifier_layer(name: str) -> bool: |
| 24 | """Check if the layer is part of classifier based on common naming patterns""" |
| 25 | classifier_keywords = ['classifier', 'fc', 'linear', 'head'] |
| 26 | return any(keyword in name.lower() for keyword in classifier_keywords) |
| 27 | |
| 28 | total_params = 0 |
| 29 | classifier_params = 0 |