Jiao77/RoRD-Layout-Recognation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change some problem 3

Browse Source
This commit is contained in:
Jiao77
2025-11-20 03:04:05 +08:00
parent 0a45856b14
commit ebda75fa5e
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tools/diffusion/ic_layout_diffusion_optimized.py
View File

@@ -233,9 +233,8 @@ class ManhattanAwareUNet(nn.Module):
nn.Conv2d(32, out_channels, 3, padding=1) nn.Conv2d(32, out_channels, 3, padding=1)
) )
# 时间融合层 # 时间融合层 - 与编码器输出通道数匹配
self.time_fusion = nn.ModuleList([ self.time_fusion = nn.ModuleList([
nn.Linear(time_dim, 64),
nn.Linear(time_dim, 128), nn.Linear(time_dim, 128),
nn.Linear(time_dim, 256), nn.Linear(time_dim, 256),
nn.Linear(time_dim, 512), nn.Linear(time_dim, 512),