fix

broadinstitute · Dec 11, 2024 · 3beef1b · 3beef1b
1 parent ab95a1e
commit 3beef1b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ml4h/models/diffusion_blocks.py b/ml4h/models/diffusion_blocks.py
@@ -209,7 +209,7 @@ def get_control_embed_model(output_maps, control_size):
     for cm in output_maps:
         control_ins.append(keras.Input(shape=cm.shape, name=cm.output_name()))
     c = layers.Concatenate()(control_ins)
-    c = layers.Dense(control_size, activation='linear')(c)
+    #c = layers.Dense(control_size, activation='linear')(c)
     return keras.Model(control_ins, c, name='control_embed')
 
 

diff --git a/ml4h/models/train.py b/ml4h/models/train.py
@@ -367,6 +367,6 @@ def train_diffusion_control_model(args):
             args.model_file = model_file
             eval_model, _, _, _ = make_multimodal_multitask_model(**args.__dict__)
             regress_on_controlled_generations(model, eval_model, tm_out, args.test_steps, args.batch_size,
-                                              0.0,2.5,f'{args.output_folder}/{args.id}/')
+                                              0.0,2.0,f'{args.output_folder}/{args.id}/')
 
     return model