inference -> O0

Mark-ZhouWX · Mark-ZhouWX · commit 53d29f766e41 · 2023-11-24T09:31:12.000+08:00
diff --git a/official/cv/segment-anything/inference.py b/official/cv/segment-anything/inference.py
@@ -81,7 +81,7 @@ def infer(args):
     )
 
     parser.add_argument("--device", type=str, default="Ascend", help="The device to run generation on.")
-    parser.add_argument("--amp_level", type=str, default="O2", help="auto mixed precision level O0, O2.")
+    parser.add_argument("--amp_level", type=str, default="O0", help="auto mixed precision level O0, O2.")
     parser.add_argument("--mode", type=int, default=0, help="MindSpore context mode. 0 for graph, 1 for pynative.")
 
     args = parser.parse_args()

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ def infer(args):`
`81`	`81`	`)`
`82`	`82`
`83`	`83`	`parser.add_argument("--device", type=str, default="Ascend", help="The device to run generation on.")`
`84`		`- parser.add_argument("--amp_level", type=str, default="O2", help="auto mixed precision level O0, O2.")`
	`84`	`+ parser.add_argument("--amp_level", type=str, default="O0", help="auto mixed precision level O0, O2.")`
`85`	`85`	`parser.add_argument("--mode", type=int, default=0, help="MindSpore context mode. 0 for graph, 1 for pynative.")`
`86`	`86`
`87`	`87`	`args = parser.parse_args()`