allenai / allenai/specter

concat_title_abstract must be True in scripts/pytorch_lightning_training_script/train.py#L73

Offen
#49 1 Kommentar 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
592
Forks
58
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

Hi @armancohan.

In [scripts/pytorch_lightning_training_script/train.py#L73](https://github.com/allenai/specter/blob/master/scripts/pytorch_lightning_training_script/train.py#L73), `concat_title_abstract` must be `True` if you want to train both with `title` and `abstract`.

However, it is setup to `None` in the actual script. That means that only the document `title` is used for training.

`concat_title_abstract: bool = None` # actual code

should be:

`concat_title_abstract: bool = True`

What do you think? Thank you.

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.