allenai / allenai/document-qa

I successfully wrote convert_to_cpu for elmo but inference takes time on cpu

Offen
#55 0 Kommentare 1 Reaktion 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
435
Forks
118
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

With the help of issues and comments reported here on convert_to_cpu.py for ELMO I wrote the script.
I am able to run the ELMO model on CPU, but the converted model is taking time for inference. Also results for GPU and CPU model is slightly different in terms of score for same trained model.

Output of convert to cpu model
```
Rebuilding
These should be close:
[False]
[array([[0.8571025]], dtype=float32)]
[array([[0.84660774]], dtype=float32)]
```

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.