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

chore: 条件编码器的激活函数改为 gelu

Browse Source
This commit is contained in:
unanmed 2026-03-31 21:01:12 +08:00
parent fee2bcf344
commit 250c2d5f67
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data/src/auto/types.ts
View File

@ -1,5 +1,3 @@
import { GinkaTopologicalGraphs } from '../topology/interface';
export const enum TowerColor {
White,
Orange,

6
ginka/maskGIT/cond.py
View File

@ -10,7 +10,7 @@ class GinkaMaskGITCond(nn.Module):
nn.Linear(cond_dim, output_dim // 2),
nn.Dropout(0.3),
nn.LayerNorm(output_dim // 2),
nn.ReLU(),
nn.GELU(),
nn.Linear(output_dim // 2, output_dim)
)
@ -18,11 +18,11 @@ class GinkaMaskGITCond(nn.Module):
self.heatmap_conv = nn.Sequential(
nn.Conv2d(heatmap_channel, output_dim // 4, kernel_size=3, padding=1, padding_mode='replicate'),
nn.BatchNorm2d(output_dim // 4),
nn.ReLU(),
nn.GELU(),
nn.Conv2d(output_dim // 4, output_dim // 2, kernel_size=3, padding=1, padding_mode='replicate'),
nn.BatchNorm2d(output_dim // 2),
nn.ReLU(),
nn.GELU(),
nn.Conv2d(output_dim // 2, output_dim, kernel_size=3, padding=1, padding_mode='replicate')
)