Skip to content

semipy.models.get_model

Warning

This section is in construction.

    semipy.models.get_model(model, num_classes)

This function checks the name of the asked model and checks if it exists in torchvision.models or if it corresponds to a Wide ResNet. It then download the architecture of the right model.

Parameters

  • model (str) - Name of the model.
  • num_classes (int) - Number of classes.