批次
This commit is contained in:
@@ -362,7 +362,7 @@ def main_dev(opt):
|
||||
precision_scope = autocast if opt.precision == "autocast" or opt.bf16 else nullcontext
|
||||
with torch.no_grad(), precision_scope(opt.device), model.ema_scope():
|
||||
#all_samples = list()
|
||||
# 执行指定的任务组数 (row)(item['number'])
|
||||
# 执行指定的任务批次 (row)(item['number'])
|
||||
for n in trange(1, desc="Sampling"):
|
||||
print("Sampling:", n)
|
||||
for prompts in tqdm(data, desc="data"):
|
||||
|
Reference in New Issue
Block a user