Update README.md

narendasan · web-flow · commit 7276ac515ab5 · 2021-08-09T10:03:07.000-06:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
    Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/README.md b/README.md
@@ -47,7 +47,7 @@ compile_settings = {
         # For static size shape=[1, 3, 224, 224]
         dtype=torch.half, # Datatype of input tensor. Allowed options torch.(float|half|int8|int32|bool)
     )],
-    "enabled_precision": {torch.half}, # Run with FP16
+    "enabled_precisions": {torch.half}, # Run with FP16
 }
 
 trt_ts_module = trtorch.compile(torch_script_module, compile_settings)

Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,7 @@ compile_settings = {`
`47`	`47`	`# For static size shape=[1, 3, 224, 224]`
`48`	`48`	`dtype=torch.half, # Datatype of input tensor. Allowed options torch.(float\|half\|int8\|int32\|bool)`
`49`	`49`	`)],`
`50`		`- "enabled_precision": {torch.half}, # Run with FP16`
	`50`	`+ "enabled_precisions": {torch.half}, # Run with FP16`
`51`	`51`	`}`
`52`	`52`
`53`	`53`	`trt_ts_module = trtorch.compile(torch_script_module, compile_settings)`