fairseqで作成されたモデルだけを,json形式で記述されたオプションから脳死で読み込んでコード内で使うためのライブラリです.
詳細はsampleフォルダへ
コマンドラインオプションではなくjson形式で扱い,純粋にmodelだけをfairseqから切り離して使いたいので作りました.
modelの改変や拡張,特殊な処理をしたデータセット読み込ませたいなど,fairseq内で完結するにはキツい作業をするときににお使いください.
詳細はrequrements.txtへ
まずfairseqのインストールを行う.https://github.com/pytorch/fairseq からリポジトリをダウンロードし,rootディレクトリで
pip install --editable ./
次に,このレポジトリをダウンロードし,rootディレクトリで
pip install --editable ./
from FairseqModelLoader import loadModel
model = loadModel(your_config_file_path)
configファイルの書き方はsample推奨
従来通りコマンドライン引数を渡しても動きます.(その場合はconfigファイルのパスを指定しないでください)
- yaminchu
- NAIST
- tktkbohshi@gmail.com
ライセンスを明示する
"hoge" is under MIT license.