(num_features, num_classes, pool_type='avg', use_conv=False)
| 32 | |
| 33 | |
| 34 | def create_classifier(num_features, num_classes, pool_type='avg', use_conv=False): |
| 35 | global_pool, num_pooled_features = _create_pool(num_features, num_classes, pool_type, use_conv=use_conv) |
| 36 | fc = _create_fc(num_pooled_features, num_classes, use_conv=use_conv) |
| 37 | return global_pool, fc |
| 38 | |
| 39 | |
| 40 | class ClassifierHead(nn.Module): |
no test coverage detected