leap-ie
HomeDashboard
v0.2.0
v0.2.0
  • Introduction
  • Quick Start
  • Installation
  • Tutorial
  • Concepts
    • Prototypes
    • Entanglements
    • Isolations
  • Integrations
  • API Reference
    • engine.generate
    • models.get_model
Powered by GitBook
On this page
  • Weights and Biases Integration
  • Hugging Face

Integrations

Weights and Biases Integration

We can also log results directly to your WandB projects. To do this, set project_name to the name of the WandB project where you'd like the results to be logged, and add your WandB API key and entity name to the config dictionary:

config = {
    "wandb_api_key": "YOUR_WANDB_API_KEY",
    "wandb_entity": "your_wandb_entity",
    "leap_api_key": "YOUR_LEAP_API_KEY",
}
df_results, dict_results = engine.generate(
    project_name="your_wandb_project_name",
    model=your_model,
    class_list=["hotdog", "not_hotdog"],
    config=config,
)

Hugging Face

We provide a convenvience function to help you download image classification and segmentation directly from Hugging Face. To do this first import the get_model function from from leap_ie.vision.models:

from leap_ie.vision.models import get_model

Then, you can download a model by passing the name of the model to get_model:

preprocessing, model, class_list = get_model("microsoft/resnet-50", source="huggingface")

For more information on the get_model function, see the .

API Reference