Merge branch 'fix-maisi' of https://github.com/KumoLiu/tutorials into…

… fix-maisi
Project-MONAI · Sep 30, 2024 · 7ab71d2 · 7ab71d2
2 parents d019799 + 97b107b
commit 7ab71d2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/generation/maisi/scripts/diff_model_create_training_data.py b/generation/maisi/scripts/diff_model_create_training_data.py
@@ -160,7 +160,9 @@ def process_file(
 
 
 @torch.inference_mode()
-def diff_model_create_training_data(env_config_path: str, model_config_path: str, model_def_path: str, num_gpus: int) -> None:
+def diff_model_create_training_data(
+    env_config_path: str, model_config_path: str, model_def_path: str, num_gpus: int
+) -> None:
     """
     Create training data for the diffusion model.
 
@@ -224,9 +226,7 @@ def diff_model_create_training_data(env_config_path: str, model_config_path: str
     parser.add_argument(
         "--model_def", type=str, default="./configs/config_maisi.json", help="Path to model definition file"
     )
-    parser.add_argument(
-        "--num_gpus", type=int, default=1, help="Number of GPUs to use for distributed training"
-    )
+    parser.add_argument("--num_gpus", type=int, default=1, help="Number of GPUs to use for distributed training")
 
     args = parser.parse_args()
     diff_model_create_training_data(args.env_config, args.model_config, args.model_def, args.num_gpus)
diff --git a/generation/maisi/scripts/diff_model_train.py b/generation/maisi/scripts/diff_model_train.py
@@ -430,9 +430,7 @@ def diff_model_train(env_config_path: str, model_config_path: str, model_def_pat
     parser.add_argument(
         "--model_def", type=str, default="./configs/config_maisi.json", help="Path to model definition file"
     )
-    parser.add_argument(
-        "--num_gpus", type=int, default=1, help="Number of GPUs to use for training"
-    )
+    parser.add_argument("--num_gpus", type=int, default=1, help="Number of GPUs to use for training")
 
     args = parser.parse_args()
     diff_model_train(args.env_config, args.model_config, args.model_def, args.num_gpus)