| 187 | } |
| 188 | |
| 189 | ResNet resnet101(int64_t num_classes) { |
| 190 | std::vector<int> layers = { 3, 4, 23, 3 }; |
| 191 | ResNet model(layers, num_classes, "resnet101"); |
| 192 | return model; |
| 193 | } |
| 194 | |
| 195 | ResNet pretrained_resnet(int64_t num_classes, std::string model_name, std::string weight_path){ |
| 196 | std::map<std::string, std::vector<int>> name2layers = getParams(); |
nothing calls this directly
no outgoing calls
no test coverage detected