fiftyone.utils.transformers.FiftyOneZeroShotTransformerForImageClassificationConfig

class documentation

class FiftyOneZeroShotTransformerForImageClassificationConfig(FiftyOneZeroShotTransformerConfig): (source)

Constructor: FiftyOneZeroShotTransformerForImageClassificationConfig(d)

View In Hierarchy

Configuration for a FiftyOneZeroShotTransformerForImageClassification.

Parameters
modela transformers model
name_or_paththe name or path to a checkpoint file to load
text_promptthe text prompt to use, e.g., "A photo of"
classesa list of custom classes for zero-shot prediction
Method __init__ Undocumented
Instance Variable name_or_path Undocumented

Inherited from FiftyOneZeroShotTransformerConfig:

Instance Variable text_prompt Undocumented

Inherited from FiftyOneTransformerConfig (via FiftyOneZeroShotTransformerConfig):

Instance Variable device Undocumented

Inherited from Config (via FiftyOneZeroShotTransformerConfig, FiftyOneTransformerConfig):

Method __repr__ Undocumented

Inherited from HasZooModel (via FiftyOneZeroShotTransformerConfig, FiftyOneTransformerConfig, Config):

Method download_model_if_necessary Undocumented
Instance Variable model_path Undocumented
Class Method _get_model Undocumented