condition and supervise

lucidtronix · lucidtronix · commit 88c59bf753ba · 2025-01-17T13:57:40.000-05:00
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

Original file line number	Diff line number	Diff line change
`@@ -302,7 +302,7 @@ def train_diffusion_control_model(args, supervised=False):`
`302`	`302`	`model = DiffusionController(`
`303`	`303`	`args.tensor_maps_in[0], args.tensor_maps_out, args.batch_size, args.dense_blocks, args.block_size, args.conv_x,`
`304`	`304`	`args.dense_layers[0], args.attention_window, args.attention_heads, args.attention_modulo, args.diffusion_loss,`
`305`		`- args.inspect_model, args.sigmoid_beta, args.diffusion_condition_strategy,`
	`305`	`+ args.sigmoid_beta, args.diffusion_condition_strategy, args.inspect_model,`
`306`	`306`	`)`
`307`	`307`
`308`	`308`	`loss = keras.losses.mean_absolute_error if args.diffusion_loss == 'mean_absolute_error' else keras.losses.mean_squared_error`