This commit is contained in:
2023-02-18 22:10:43 +08:00
parent 53cc9d52d3
commit bab2a25c96

View File

@@ -241,6 +241,7 @@ def main_dev(opt):
opt.prompt = item['prompt'] # 描述 opt.prompt = item['prompt'] # 描述
if 'number' in item: if 'number' in item:
opt.n_samples = item['number'] # 列数 opt.n_samples = item['number'] # 列数
print(f"n_samples: {opt.n_samples}")
#if 'n_rows' in item: #if 'n_rows' in item:
# opt.n_rows = item['n_rows'] # 行数 # opt.n_rows = item['n_rows'] # 行数
if 'scale' in item: if 'scale' in item:
@@ -364,7 +365,7 @@ def main_dev(opt):
#all_samples = list() #all_samples = list()
# 执行指定的任务批次 (row)(item['number']) # 执行指定的任务批次 (row)(item['number'])
for n in trange(1, desc="Sampling"): for n in trange(1, desc="Sampling"):
print("Sampling:", n) print("Sampling:", data)
for prompts in tqdm(data, desc="data"): for prompts in tqdm(data, desc="data"):
uc = None uc = None
if opt.scale != 1.0: if opt.scale != 1.0: