MultiModalPredictor.list_supported_models#

MultiModalPredictor.list_supported_models(pretrained=True)[source]#

List supported models for each problem_type to let users know options of checkpoint name to choose during fit().

Parameters

pretrained (bool, default = True) – If True, only return the models with pretrained weights. If False, return all the models as long as there is model definition.

Return type

a list of model names