Commit
•
2f6d761
1
Parent(s):
57f68e5
Add model
Browse files- README.md +2 -0
- definition.json +1 -1
README.md
CHANGED
|
@@ -17,6 +17,8 @@ The model can be loaded with [experimaestro
|
|
| 17 |
IR](https://experimaestro-ir.readthedocs.io/en/latest/)
|
| 18 |
|
| 19 |
```py from xpmir.models import AutoModel
|
|
|
|
|
|
|
| 20 |
# Model that can be re-used in experiments
|
| 21 |
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_DistilMSE")
|
| 22 |
|
|
|
|
| 17 |
IR](https://experimaestro-ir.readthedocs.io/en/latest/)
|
| 18 |
|
| 19 |
```py from xpmir.models import AutoModel
|
| 20 |
+
from xpmir.models import AutoModel
|
| 21 |
+
|
| 22 |
# Model that can be re-used in experiments
|
| 23 |
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_DistilMSE")
|
| 24 |
|
definition.json
CHANGED
|
@@ -1 +1 @@
|
|
| 1 |
-
[{"id":
|
|
|
|
| 1 |
+
[{"id": 140408891192656, "module": "xpmir.text.huggingface", "type": "TransformerTokensEncoder", "typename": "xpmir.text.huggingface.TransformerTokensEncoder", "identifier": "d26c7c4d00bb2002f07f80036294ed81355c1256fde3758c6ab9451c7fc06c2c", "fields": {"model_id": "distilbert-base-uncased", "trainable": true, "layer": 0, "dropout": 0.0}}, {"id": 140408891192224, "module": "xpmir.neural.splade", "type": "MaxAggregation", "typename": "xpmir.neural.splade.MaxAggregation", "identifier": "6d85ac05decb4f139354ac3f79990a2afdb8ced21c8861f2dd5c720484514f48", "fields": {}}, {"id": 140408891193088, "module": "xpmir.neural.splade", "type": "SpladeTextEncoder", "typename": "xpmir.neural.splade.SpladeTextEncoder", "identifier": "9f455ba2eb9c3dc5ce40d4f0620b24a8d086b1a35449380af6cf625a69f2efeb", "fields": {"encoder": {"type": "python", "value": 140408891192656}, "aggregation": {"type": "python", "value": 140408891192224}, "maxlen": 200}}, {"id": 140408891195056, "module": "xpmir.neural.splade", "type": "SpladeTextEncoder", "typename": "xpmir.neural.splade.SpladeTextEncoder", "identifier": "af02fa8dcdc136f3c02eeb124fe524302c85d736deeaa6d04b3acc215d1593ac", "fields": {"encoder": {"type": "python", "value": 140408891192656}, "aggregation": {"type": "python", "value": 140408891192224}, "maxlen": 30}}, {"id": 140408891390704, "module": "xpmir.neural.dual", "type": "DotDense", "typename": "xpmir.neural.dual.DotDense", "identifier": "d4e16bd0471b8b48c996614c52a927c911296acf20fca61b75b431499cf47b47", "fields": {"checkpoint": {"type": "path.serialized", "value": "checkpoint", "is_folder": true}, "encoder": {"type": "python", "value": 140408891193088}, "query_encoder": {"type": "python", "value": 140408891195056}}}]
|