mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-27 23:47:46 +00:00
Tim
35c209b092
* feat: start with previous workspace * feat: finetune llm * add PrevModelLoadEvaluator --------- Co-authored-by: Young <afe.young@gmail.com> Co-authored-by: v-jianwan <v-jianwan@microsoft.com> Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import asyncio
|
|
from pathlib import Path
|
|
|
|
import fire
|
|
|
|
from rdagent.app.data_science.conf import DS_RD_SETTING
|
|
from rdagent.app.finetune.data_science.conf import update_settings
|
|
from rdagent.core.utils import import_class
|
|
from rdagent.log import rdagent_logger as logger
|
|
from rdagent.scenarios.data_science.loop import DataScienceRDLoop
|
|
|
|
|
|
def main(
|
|
model: str | None = None,
|
|
competition: str | None = None,
|
|
):
|
|
"""
|
|
Parameters
|
|
----------
|
|
competition :
|
|
Competition name.
|
|
|
|
Auto R&D Evolving loop for models finetune.
|
|
You can continue running a session by using the command:
|
|
.. code-block:: bash
|
|
dotenv run -- python rdagent/app/finetune/data_science/loop.py --competition aerial-cactus-identification
|
|
"""
|
|
if not competition:
|
|
raise Exception("Please specify competition name.")
|
|
|
|
model_folder = Path(DS_RD_SETTING.local_data_path) / competition / "prev_model"
|
|
if not model_folder.exists():
|
|
raise Exception(f"Please put the model path to {model_folder}.")
|
|
update_settings(competition)
|
|
rd_loop: DataScienceRDLoop = DataScienceRDLoop(DS_RD_SETTING)
|
|
asyncio.run(rd_loop.run())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
fire.Fire(main)
|