std var

kengz · Dec 22, 2024 · 30bfe89 · 30bfe89
1 parent 1a3b3d2
commit 30bfe89
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/v2/module/merge.py b/v2/module/merge.py
@@ -62,10 +62,10 @@ class FiLMMerge(nn.Module):
     FiLM(x) = gamma(z) * x + beta(z)
     """
 
-    def __init__(self, feature_size: int, conditioner_size: int):
+    def __init__(self, feature_dim: int, conditioner_dim: int):
         super().__init__()
-        self.gamma = nn.Linear(conditioner_size, feature_size)
-        self.beta = nn.Linear(conditioner_size, feature_size)
+        self.gamma = nn.Linear(conditioner_dim, feature_dim)
+        self.beta = nn.Linear(conditioner_dim, feature_dim)
 
     def forward(self, feature: torch.Tensor, conditioner: torch.Tensor) -> torch.Tensor:
         return self.gamma(conditioner) * feature + self.beta(conditioner)
diff --git a/v2/specs/merge/film.yaml b/v2/specs/merge/film.yaml
@@ -9,8 +9,8 @@ modules:
       out_features: 4
   merge_1_2:
     MergeFiLM:
-      feature_size: 10
-      condition_size: 4
+      feature_dim: 10
+      condition_dim: 4
   tail:
     Linear:
       in_features: 10