From 389eb180589d2f672163f76552cc3c90e54dd6fe Mon Sep 17 00:00:00 2001 From: MaybeRichard Date: Sat, 22 Mar 2025 17:16:55 +0800 Subject: [PATCH 1/2] Fix typos and bugs --- generation/maisi/scripts/diff_model_train.py | 4 ++-- generation/maisi/scripts/train_controlnet.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/generation/maisi/scripts/diff_model_train.py b/generation/maisi/scripts/diff_model_train.py index c616b89c3..998f8304a 100644 --- a/generation/maisi/scripts/diff_model_train.py +++ b/generation/maisi/scripts/diff_model_train.py @@ -480,13 +480,13 @@ def diff_model_train( parser.add_argument( "--env_config", type=str, - default="./configs/environment_maisi_diff_model_train.json", + default="./configs/environment_maisi_diff_model.json", help="Path to environment configuration file", ) parser.add_argument( "--model_config", type=str, - default="./configs/config_maisi_diff_model_train.json", + default="./configs/config_maisi_diff_model.json", help="Path to model training/inference configuration", ) parser.add_argument( diff --git a/generation/maisi/scripts/train_controlnet.py b/generation/maisi/scripts/train_controlnet.py index 3d7336be1..491afc6cb 100644 --- a/generation/maisi/scripts/train_controlnet.py +++ b/generation/maisi/scripts/train_controlnet.py @@ -43,7 +43,7 @@ def main(): parser.add_argument( "-c", "--config-file", - default="./configs/config_maisi.json", + default="./configs/config_maisi-ddpm.json", help="config json file that stores network hyper-parameters", ) parser.add_argument( @@ -269,7 +269,7 @@ def main(): for label in weighted_loss_label: roi[interpolate_label == label] = 1 weights[roi.repeat(1, images.shape[1], 1, 1, 1) == 1] = weighted_loss - loss = (F.l1_loss(noise_pred.float(), model_gt.float(), reduction="none") * weights).mean() + loss = (F.l1_loss(model_output.float(), model_gt.float(), reduction="none") * weights).mean() else: loss = F.l1_loss(model_output.float(), model_gt.float()) From e05e3ca52342e1bf5e5f8923a3a3308e97987d1d Mon Sep 17 00:00:00 2001 From: MaybeRichard Date: Thu, 3 Apr 2025 09:18:42 +0800 Subject: [PATCH 2/2] DCO Remediation Commit for MaybeRichard I, MaybeRichard , hereby add my Signed-off-by to this commit: 389eb180589d2f672163f76552cc3c90e54dd6fe Signed-off-by: MaybeRichard