SPLADE_DistilMSE / definition.json
yzong12138's picture
Add model
aff456e
raw
history blame
1.72 kB
[{"id": 139993107703040, "module": "xpmir.text.huggingface", "type": "TransformerVocab", "identifier": "1b34882d3bb2bdecb562df9dbc0611309178933028292b272c8d2b6121e5f6b2", "typename": "xpmir.text.huggingface.TransformerVocab", "ignored": [], "fields": {"model_id": "distilbert-base-uncased", "trainable": true, "layer": 0, "dropout": 0.0}}, {"id": 139993107702992, "module": "xpmir.neural.splade", "type": "MaxAggregation", "identifier": "6d85ac05decb4f139354ac3f79990a2afdb8ced21c8861f2dd5c720484514f48", "typename": "xpmir.neural.splade.MaxAggregation", "ignored": [], "fields": {}}, {"id": 139993107703472, "module": "xpmir.neural.splade", "type": "SpladeTextEncoder", "identifier": "e54501ae9b19387204445bec69bc5a860d9e68f1e44d75733fe66b4e02418098", "typename": "xpmir.neural.splade.SpladeTextEncoder", "ignored": [], "fields": {"encoder": {"type": "python", "value": 139993107703040}, "aggregation": {"type": "python", "value": 139993107702992}, "maxlen": 200}}, {"id": 139993107704720, "module": "xpmir.neural.splade", "type": "SpladeTextEncoder", "identifier": "1e5b132f4b528e408796f70295156b3ad0d6c41f669f2b43a5a1e3ab5ac7906e", "typename": "xpmir.neural.splade.SpladeTextEncoder", "ignored": [], "fields": {"encoder": {"type": "python", "value": 139993107703040}, "aggregation": {"type": "python", "value": 139993107702992}, "maxlen": 30}}, {"id": 139993107853856, "module": "xpmir.neural.dual", "type": "DotDense", "identifier": "f4bacc4feb44bae72cbe080cad5b588a32de094a89fd4524520842e342a1a48d", "fields": {"checkpoint": {"type": "path.serialized", "value": "checkpoint", "is_folder": true}, "encoder": {"type": "python", "value": 139993107703472}, "query_encoder": {"type": "python", "value": 139993107704720}}}]