condition and supervise

broadinstitute · Jan 17, 2025 · 88c59bf · 88c59bf
1 parent 21039b1
commit 88c59bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4h/models/train.py b/ml4h/models/train.py
@@ -302,7 +302,7 @@ def train_diffusion_control_model(args, supervised=False):
         model = DiffusionController(
             args.tensor_maps_in[0], args.tensor_maps_out, args.batch_size, args.dense_blocks, args.block_size, args.conv_x,
             args.dense_layers[0], args.attention_window, args.attention_heads, args.attention_modulo, args.diffusion_loss,
-            args.inspect_model, args.sigmoid_beta, args.diffusion_condition_strategy,
+            args.sigmoid_beta, args.diffusion_condition_strategy, args.inspect_model,
         )
 
     loss = keras.losses.mean_absolute_error if args.diffusion_loss == 'mean_absolute_error' else keras.losses.mean_squared_error