简体中文
Tutorial 中文教程 Demo Video
NeuronBlocks ist ein NLP Deep Learning Modeling Toolkit , mit dem Ingenieure/Forscher End-to-End-Pipelines für das Modell der neuronalen Netzwerkmodelltraining für NLP-Aufgaben erstellen können. Das Hauptziel dieses Toolkits ist es, die Entwicklungskosten für das Gebäudetodellaufbau des NLP Deep Neural Network zu minimieren, einschließlich der Trainings- und Inferenzphasen.
NeuronBlocks besteht aus zwei Hauptkomponenten: Blockzoo und Modellzoo .
Benutzer können entweder vorhandene Modelle (Konfigurationsdateien) im Modellzoo auswählen, um das Modelltraining zu starten oder neue Modelle zu erstellen, indem sie neuronale Netzwerkblöcke im Blockzoo genutzt, genau wie das Spielen mit LEGO.
Hinweis: NeuronBlocks erfordert Python 3.6 und höher.
Klonen Sie dieses Projekt.
git clone https://github.com/Microsoft/NeuronBlocksInstallieren Sie Python -Pakete in Anforderungen.txt unter dem folgenden Befehl.
pip install -r requirements.txtInstallieren Sie Pytorch ( NeuronBlocks unterstützt Pytorch 0.4.1 und höher ).
Führen Sie für Linux den folgenden Befehl aus:
pip install " torch>=0.4.1 "Für Windows empfehlen wir Ihnen, Pytorch über Conda zu installieren, indem Sie die Anweisung von Pytorch befolgen.
Beginnen Sie mit den angegebenen Beispielen. Sowohl Linux/Windows, GPU/CPU werden unterstützt. Für Windows empfehlen wir Ihnen, PowerShell anstelle von CMD zu verwenden.
Tipps: In der folgenden Anweisung bezeichnet Projectroot das Stammverzeichnis dieses Projekts.
# train
cd PROJECT_ROOT
python train.py --conf_path=model_zoo/demo/conf.json
# test
python test.py --conf_path=model_zoo/demo/conf.json
# predict
python predict.py --conf_path=model_zoo/demo/conf.jsonFür die Vorhersage haben Neuronblocks zwei Modi: interaktiv und Charge .
# use the above example
# interactive prediction
python predict.py --conf_path=model_zoo/demo/conf.json --predict_mode= ' interactive ' # use the above example
# batch prediction
python predict.py --conf_path=model_zoo/demo/conf.json --predict_mode= ' batch ' --predict_data_path=dataset/demo/predict.tsvWeitere Informationen finden Sie in Tutorial.MD und Code -Dokumentation.
Ingenieure oder Forscher, die sich vor den folgenden Herausforderungen stellen, wenn sie neuronale Netzwerkmodelle verwenden, um NLP -Probleme anzugehen:
Die Vorteile der Nutzung von Neuronblocks für das NLP -Modell der neuronalen Netzwerk -Modells umfassen:
Modellbildung : Für das Modellieren und Parameterabstieg müssen Benutzer nur einfache JSON -Konfigurationsdateien schreiben, die den Aufwand für die Implementierung neuer Ideen erheblich minimieren.
Modellfreigabe Es ist super einfach, Modelle genau über JSON -Dateien zu teilen, anstatt durch böse Codes. Bei verschiedenen Modellen oder Aufgaben müssen unsere Benutzer nur eine einzelne zentralisierte Quellcode -Basis beibehalten.
Wiederverwendbarkeit von Code : Gemeinsame Blöcke können einfach über verschiedene Modelle oder Aufgaben ausgetauscht werden, wodurch die doppelte Codierungsarbeiten reduziert werden.
Plattformflexibilität : NeuronBlocks können sowohl auf Linux- als auch auf Windows -Maschinen mit CPU und GPU ausgeführt werden. Es unterstützt auch Schulungen auf GPU -Plattformen wie Philly und Pai.
| CPU -Inferenz | Einzel-GPU-Inferenz | Multi-GPU-Inferenz | |
| CPU -Zug | ✓ | ✓ | ✓ |
| Ein-GPU-Zug | ✓ | ✓ | ✓ |
| Multi-GPU-Zug | ✓ | ✓ | ✓ |
Modellvisualisierung : Ein Modellvisualisierer wird zur Visualisierung und Konfiguration der Korrektheitsprüfung bereitgestellt, mit der Benutzer die Modellarchitektur während des Debuggens leicht visualisieren können.
Erweiterbarkeit : NeuronBlocks ist erweiterbar, sodass Benutzer neue Blöcke beitragen oder neuartige Modelle (JSON -Dateien) beitragen können.
NeuronBlocks arbeitet in einem offenen Modell. Es wurde von der STCA NLP Group, Microsoft , entwickelt und entwickelt. Beiträge aus Akademie und Industrie sind ebenfalls sehr willkommen. Weitere Informationen finden Sie unter Beitrags.md.
Jeder, der vertraut ist, ist sehr ermutigt, Code beizutragen.
NeuronBlocks - Erstellen Sie Ihre NLP -DNN -Modelle wie das Spielen von LEGO. EMNLP 2019 , unter https://arxiv.org/abs/1904.09535.
@article{gong2019neuronblocks,
title={NeuronBlocks--Building Your NLP DNN Models Like Playing Lego},
author={Gong, Ming and Shou, Linjun and Lin, Wutao and Sang, Zhijie and Yan, Quanjia and Yang, Ze, Cheng, Feixiang and Jiang, Daxin},
journal={arXiv preprint arXiv:1904.09535},
year={2019}
}
Copyright (C) Microsoft Corporation. Alle Rechte vorbehalten.
Lizenziert unter der MIT -Lizenz.
Wenn Sie Fragen haben, wenden Sie sich bitte an [email protected]
Wenn Sie WeChat haben, können Sie auch das folgende Konto hinzufügen: