From ebda75fa5e729c68e230f4cdf50179df046ea1b4 Mon Sep 17 00:00:00 2001 From: Jiao77 Date: Thu, 20 Nov 2025 03:04:05 +0800 Subject: [PATCH] change some problem 3 --- tools/diffusion/ic_layout_diffusion_optimized.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tools/diffusion/ic_layout_diffusion_optimized.py b/tools/diffusion/ic_layout_diffusion_optimized.py index b631ee4..98064c4 100644 --- a/tools/diffusion/ic_layout_diffusion_optimized.py +++ b/tools/diffusion/ic_layout_diffusion_optimized.py @@ -233,9 +233,8 @@ class ManhattanAwareUNet(nn.Module): nn.Conv2d(32, out_channels, 3, padding=1) ) - # 时间融合层 + # 时间融合层 - 与编码器输出通道数匹配 self.time_fusion = nn.ModuleList([ - nn.Linear(time_dim, 64), nn.Linear(time_dim, 128), nn.Linear(time_dim, 256), nn.Linear(time_dim, 512),