mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-28 16:07:46 +00:00

Files

T

Xinjie Shen f20dc4482b example workflow code for model implementation (#17 )

* upload example code for model implementation

* Refactor Model Implement

* add export

---------

Co-authored-by: Young <afe.young@gmail.com>

2024-06-21 16:41:34 +08:00

eval.py

2024-06-21 16:41:34 +08:00

README.md

2024-06-21 16:41:34 +08:00

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

From paper to task.

python rdagent/app/model_implementation/task_extraction.py
# It may based on rdagent/document_reader/document_reader.py

From paper to implementation

# Similar to
# rdagent/app/factor_extraction_and_implementation/factor_extract_and_implement.py

python rdagent/app/model_implementation/eval.py

TODO:
- Is evaluation reasonable