condition strategy

broadinstitute · Dec 16, 2024 · a3165b6 · a3165b6
1 parent 4e7cebb
commit a3165b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ml4h/models/diffusion_blocks.py b/ml4h/models/diffusion_blocks.py
@@ -103,8 +103,8 @@ def apply(x):
 
 def condition_layer_film(input_tensor, control_vector, filters):
     # Transform control into gamma and beta
-    gamma = layers.Dense(filters, activation="linear")(control_vector)
-    beta = layers.Dense(filters, activation="linear")(control_vector)
+    gamma = layers.Dense(filters*2, activation="linear")(control_vector)
+    beta = layers.Dense(filters*2, activation="linear")(control_vector)
 
     # Reshape gamma and beta to match the spatial dimensions
     #gamma = tf.reshape(gamma, (-1,) + input_tensor.shape[1:-1] + (filters,))