mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-08 03:10:24 +00:00
set num_runs=1
This commit is contained in:
parent
fd2f4d5e37
commit
890acbffc9
1 changed files with 5 additions and 6 deletions
|
|
@ -74,20 +74,19 @@ def run_agent(dataset_name: str, experiment_suffix: str, max_workers: int, num_r
|
|||
|
||||
|
||||
def main():
|
||||
max_workers = 6
|
||||
num_runs = 4 # Run each task 4 times
|
||||
max_workers = 8
|
||||
num_runs = 1 # Run each task 4 times
|
||||
if max_workers > 1:
|
||||
ray.init(num_cpus=6)
|
||||
# run_agent(dataset_name="train", experiment_suffix="v2", max_workers=max_workers, num_runs=num_runs)
|
||||
ray.init(num_cpus=8)
|
||||
|
||||
logger.info("Start running experiments without experience")
|
||||
for i in range(num_runs):
|
||||
run_agent(dataset_name="dev", experiment_suffix=f"no-exp", max_workers=max_workers, num_runs=1,
|
||||
use_experience=False, workspace_id="appworld_8b_0725")
|
||||
use_experience=False, workspace_id="appworld_v1")
|
||||
|
||||
logger.info("Start running experiments with experience")
|
||||
for i in range(num_runs):
|
||||
run_agent(dataset_name="dev", experiment_suffix=f"add-exp", max_workers=max_workers, num_runs=1, use_experience=True,workspace_id="appworld_8b_0725")
|
||||
run_agent(dataset_name="dev", experiment_suffix=f"add-exp", max_workers=max_workers, num_runs=1, use_experience=True,workspace_id="appworld_v1")
|
||||
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue