Merge branch 'main' of github.com:CompNet/Tibert

Author: Aethor

README.md

@@ -151,3 +151,17 @@ The following table presents the results we obtained by training this model (for
 | Dataset | Base model        | MUC   | B3    | CEAF  | CoNLL F1 |
 |---------|-------------------|-------|-------|-------|----------|
 | Litbank | `bert-base-cased` | 77.35 | 67.63 | 56.66 | 67.21    |
+
+
+# Citation
+
+If you use this software in a research project, you can cite Tibert as follows:
+
+```bibtex
+@Misc{tibert,
+  author = {Amalvy, A. and Labatut, V. and Dufour, R.},
+  title = {Tibert},
+  year  = {2023},
+  url   = {https://github.com/CompNet/Tibert},
+}
+```

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tibert"
-version = "0.1.0"
+version = "0.2.0"
 description = "BERT for Coreference Resolution"
 authors = ["Arthur Amalvy <[email protected]>"]
 license = "GPL-3.0-only"

tibert/run_train.py

@@ -103,6 +103,7 @@ def main(
         bert_lr,
         task_lr,
         out_model_path,
+        "auto",
         _run,
     )