From 0b4545bdd47186e03fb9bc4d0e74ca89922aa5a6 Mon Sep 17 00:00:00 2001 From: Boris Fomitchev Date: Thu, 19 Dec 2024 18:11:22 -0800 Subject: [PATCH] cleanup Signed-off-by: Boris Fomitchev --- generation/maisi/scripts/inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/generation/maisi/scripts/inference.py b/generation/maisi/scripts/inference.py index 6fa85e7b3..968d5bf49 100644 --- a/generation/maisi/scripts/inference.py +++ b/generation/maisi/scripts/inference.py @@ -128,7 +128,7 @@ def main(): print(f"{k}: {val}") print("Global config variables have been loaded.") - # ## Read in configuration setting, including network definition, body region and anatomy to generate, etc. + # ## Read in configuration setting, including network definition, body region and anatomy to generate, etc. # # The information for the inference input, like body region and anatomy to generate, is stored in "./configs/config_infer.json". # Please refer to README.md for the details. @@ -141,7 +141,7 @@ def main(): # override num_split if asked if "autoencoder_tp_num_splits" in config_infer_dict: args.autoencoder_def["num_splits"] = config_infer_dict["autoencoder_tp_num_splits"] - args.mask_generation_autoencoder_def["num_splits"] = config_infer_dict["autoencoder_tp_num_splits"] + args.mask_generation_autoencoder_def["num_splits"] = config_infer_dict["autoencoder_tp_num_splits"] for k, v in config_infer_dict.items(): setattr(args, k, v) print(f"{k}: {v}")