mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 16:07:46 +00:00
6b626eb56d
* Commit init framework * Co-authored-by: Yuante Li (FESCO Adecco Human Resources) <v-yuanteli@microsoft.com> Co-authored-by: XianBW <XianBW@users.noreply.github.com> * add an import * refine the whole framework * benchmark related framework * fix black and isort errors * move requirements to folder * fix black again --------- Co-authored-by: Young <afe.young@gmail.com> Co-authored-by: xuyang1 <xuyang1@microsoft.com>
Preparation
Install Pytorch
CPU CUDA will be enough for verify the implementation
Please install pytorch based on your system. Here is an example on my system
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip3 install torch_geometric
Tasks
Task Extraction
From paper to task.
python rdagent/app/model_implementation/task_extraction.py
# It may based on rdagent/document_reader/document_reader.py
Complete workflow
From paper to implementation
# Similar to
# rdagent/app/factor_extraction_and_implementation/factor_extract_and_implement.py
Paper benchmark
python rdagent/app/model_implementation/eval.py
TODO:
- Is evaluation reasonable