fiftyone.utils.transformers.FiftyOneZeroShotTransformerConfig

class documentation

class FiftyOneZeroShotTransformerConfig(FiftyOneTransformerConfig): (source)

Known subclasses: fiftyone.utils.transformers.FiftyOneZeroShotTransformerForImageClassificationConfig, fiftyone.utils.transformers.FiftyOneZeroShotTransformerForObjectDetectionConfig

Constructor: FiftyOneZeroShotTransformerConfig(d)

View In Hierarchy

Configuration for a FiftyOneZeroShotTransformer.

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 classes Undocumented
Instance Variable text_prompt Undocumented

Inherited from FiftyOneTransformerConfig:

Instance Variable device Undocumented

Inherited from Config (via FiftyOneTransformerConfig):

Method __repr__ Undocumented

Inherited from HasZooModel (via FiftyOneTransformerConfig, Config):

Method download_model_if_necessary Undocumented
Instance Variable model_path Undocumented
Class Method _get_model Undocumented