Skip to content

Evaluation Parameters

Required Evaluation Parameters

Name Type Default value Description
batchSize int 64 Batch size per GPU.
numWorkers int 1 Number of data loader workers per GPU.
leftKeys list[str] ['text'] List of column names for the left features, typically text fields.
rightKeys list[str] ['image_local', 'title'] List of column names for the right features. The first field should be an image, and the others should be text fields.
imgOrTxt str [["txt"], ["img", "txt"]] CSV column name used to represent weights.
leftWeights list[str] "" List of weights for the left features.
rightWeights list[float] "" List of weights for the right features.
contextLength str "" List (represented as a string) that indicates the context length.

Example

evaluate_task_params = {
    "batch-size": 32,
    "num_workers": 4,
    "left-keys": ["query"],
    "right-keys": ["my_image", "my_text"],
    "img-or-txt": [["txt"], ["img", "txt"]],
    "left-weights": [1],
    "right-weights": [0.9, 0.1],
    "context-length": [[77], [0, 77]]
}