condition and supervise

broadinstitute · Jan 13, 2025 · 1ca98be · 1ca98be
1 parent 5cd6e08
commit 1ca98be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4h/models/train.py b/ml4h/models/train.py
@@ -376,7 +376,7 @@ def train_diffusion_control_model(args, supervised=False):
         data, labels, paths = big_batch_from_minibatch_generator(generate_test, 1)
         preds = model.plot_reconstructions((data, labels), prefix=f'{args.output_folder}/{args.id}/reconstructions/')
         image_out = {args.tensor_maps_in[0].output_name(): data[args.tensor_maps_in[0].input_name()]}
-        predictions_to_pngs(preds, args.tensor_maps_in, args.tensor_maps_in, data, image_out, paths, f'{args.output_folder}/{args.id}/')
+        predictions_to_pngs(preds, args.tensor_maps_in, args.tensor_maps_in, data, image_out, paths, f'{args.output_folder}/{args.id}/reconstructions/')
         interpolate_controlled_generations(model, args.tensor_maps_out, args.tensor_maps_out[0], args.batch_size,
                                            f'{args.output_folder}/{args.id}/')
         if model.input_map.axes() == 2: