gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-08 20:47:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Build model class inheritance (#44)

Browse Source 
* update all code

* fix a typo

---------

Co-authored-by: xuyang1 <xuyang1@microsoft.com>
This commit is contained in:
Xu Yang
2024-07-03 17:42:07 +08:00
committed by GitHub
parent ab2f61fe9d
commit d88ddc0c1c
8 changed files with 138 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/app/model_implementation/eval.py
+5 -5
View File
@@ -5,13 +5,13 @@ DIRNAME = Path(__file__).absolute().resolve().parent
from rdagent.components.task_implementation.model_implementation.benchmark.eval import (
ModelImpValEval,
)
from rdagent.components.task_implementation.model_implementation.one_shot import (
ModelTaskGen,
)
from rdagent.components.task_implementation.model_implementation.task import (
from rdagent.components.task_implementation.model_implementation.model import (
ModelImpLoader,
ModelTaskLoaderJson,
)
from rdagent.components.task_implementation.model_implementation.one_shot import (
ModelCodeWriter,
)
bench_folder = DIRNAME.parent.parent / "components" / "task_implementation" / "model_implementation" / "benchmark"
mtl = ModelTaskLoaderJson(str(bench_folder / "model_dict.json"))
@@ -20,7 +20,7 @@ task_l = mtl.load()
task_l = [t for t in task_l if t.key == "A-DGN"] # FIXME: other models does not work well
mtg = ModelTaskGen()
mtg = ModelCodeWriter()
impl_l = mtg.generate(task_l)