onnx/train_config.json

{
    "lr": "0.0003",
    "prompt-query": "none",
    "attribute": "description",
    "epochs": "5",
    "train-type": "PosPair",
    "loss": "MultipleNegativesRankingLoss",
    "metric": "InformationRetrievalEvaluator",
    "trainable-layer": "6",
    "newmodel-name": "Algolia/Algolia-large-en-generic-v2410_ONNX",
    "basemodel-name": "gs://models-experimental/finetuned-models/Algolia/Algolia-large-en-title-trainhomedepot-devhomedepot-15036",
    "batch-size": "1024",
    "bq-uri": "bq://data-science-365419.datasets_experimental.homedepot_en_train",
    "bq-uri-dev": "bq://data-science-365419.datasets_experimental.homedepot_en_dev",
    "modelname": "thenlper/gte-large",
    "project-id": "data-science-365419"
}