MCPcopy Create free account
hub / github.com/AllentDan/LibtorchTutorials / resnet101

Function resnet101

lesson6-Segmentation/ResNet.cpp:189–193  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

187}
188
189ResNet 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
195ResNet pretrained_resnet(int64_t num_classes, std::string model_name, std::string weight_path){
196 std::map<std::string, std::vector<int>> name2layers = getParams();

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected