(self, num_classes, global_pool='')
| 379 | return self.head |
| 380 | |
| 381 | def reset_classifier(self, num_classes, global_pool=''): |
| 382 | self.num_classes = num_classes |
| 383 | self.head = nn.Linear(self.embed_dim, num_classes) if num_classes > 0 else nn.Identity() |
| 384 | |
| 385 | def inflate_weight(self, weight_2d, time_dim, center=False): |
| 386 | if center: |
nothing calls this directly
no outgoing calls
no test coverage detected