Upload tarte_pretrained_configs
Browse files
tarte_pretrained_configs.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"data_kg_dir": "/data/parietal/store3/soda/felefebv/gitlab/tarte/data/yago4.5_enriched/processed", "data_name": "yago4.5_enriched", "num_steps": 200000, "num_warm_steps": 2000, "learning_rate": 1e-06, "dim_input": 300, "dim_transformer": 768, "dim_feedforward": 2048, "dim_projection": [64, 128, 256, 512, 768], "num_heads": 24, "num_layers_transformer": 3, "dropout": 0.1, "num_hops": 1, "num_entity": 512, "num_pos": 1, "min_rels": 3, "max_num_per_rel": 1, "loss_terms": "infonce"}
|