AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-19 16:41:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: 验证

Browse Source
This commit is contained in:
unanmed 2026-03-11 22:57:47 +08:00
parent a0faada62b
commit 36265a9bce
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ginka/train_maskGIT.py
View File

@ -196,7 +196,7 @@ def train():
idx += 1 idx += 1
# 2. 从头完整生成 # 2. 从头完整生成
map = torch.full((1, MAP_SIZE), MASK_TOKEN).to(device) map = torch.full((B, MAP_SIZE), MASK_TOKEN).to(device)
for i in range(GENERATE_STEP): for i in range(GENERATE_STEP):
# 1. 预测 # 1. 预测
logits = model(map, cond, heatmap) # [1, H * W, num_classes] logits = model(map, cond, heatmap) # [1, H * W, num_classes]
@ -223,7 +223,7 @@ def train():
if (map == MASK_TOKEN).sum() == 0: if (map == MASK_TOKEN).sum() == 0:
break break
generated_img = matrix_to_image_cv(map.view(1, H, W)[0].cpu().numpy(), tile_dict) generated_img = matrix_to_image_cv(map.view(B, H, W)[0].cpu().numpy(), tile_dict)
cv2.imwrite(f"result/transformer_img/g-{idx}.png", generated_img) cv2.imwrite(f"result/transformer_img/g-{idx}.png", generated_img)
avg_loss_val = val_loss_total.item() / len(dataloader_val) avg_loss_val = val_loss_total.item() / len(dataloader_val)