diff --git a/data/src/auto/types.ts b/data/src/auto/types.ts index 256f8f1..4ec1a32 100644 --- a/data/src/auto/types.ts +++ b/data/src/auto/types.ts @@ -1,5 +1,3 @@ -import { GinkaTopologicalGraphs } from '../topology/interface'; - export const enum TowerColor { White, Orange, diff --git a/ginka/maskGIT/cond.py b/ginka/maskGIT/cond.py index b75c377..af45eca 100644 --- a/ginka/maskGIT/cond.py +++ b/ginka/maskGIT/cond.py @@ -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') )