update

guopengf · guopengf · commit 5884e1917870 · 2024-07-02T23:48:11.000Z
Signed-off-by: Pengfei Guo &lt;pengfeig@nvidia.com&gt;
diff --git a/generative/maisi/scripts/train_controlnet.py b/generative/maisi/scripts/train_controlnet.py
@@ -170,7 +170,7 @@ def main():
         nargs="+",
         default=[],
         action="store_true",
-        help="list of lables that use weighted loss",
+        help="list of lables that apply weighted loss",
     )
     parser.add_argument("-l", "--weighted_loss", default=100, type=int, help="loss weight loss for ROI labels")
     args = parser.parse_args()

Original file line number	Diff line number	Diff line change
`@@ -170,7 +170,7 @@ def main():`
`170`	`170`	`nargs="+",`
`171`	`171`	`default=[],`
`172`	`172`	`action="store_true",`
`173`		`- help="list of lables that use weighted loss",`
	`173`	`+ help="list of lables that apply weighted loss",`
`174`	`174`	`)`
`175`	`175`	`parser.add_argument("-l", "--weighted_loss", default=100, type=int, help="loss weight loss for ROI labels")`
`176`	`176`	`args = parser.parse_args()`